Update retrieve.py

arnavsinghvi11 · web-flow · commit bb97c72da631 · 2024-06-15T11:26:52.000-07:00
ruff fix
diff --git a/dspy/retrieve/retrieve.py b/dspy/retrieve/retrieve.py
@@ -70,7 +70,7 @@ def forward(
             return Prediction(passages=passages)
         else:
             passages = dsp.retrieveEnsemblewithMetadata(
-                queries, k=k, by_prob=by_prob, **kwargs
+                queries, k=k, by_prob=by_prob, **kwargs,
             )
             if isinstance(passages[0], List):
                 pred_returns = []

Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ def forward(`
`70`	`70`	`return Prediction(passages=passages)`
`71`	`71`	`else:`
`72`	`72`	`passages = dsp.retrieveEnsemblewithMetadata(`
`73`		`- queries, k=k, by_prob=by_prob, **kwargs`
	`73`	`+ queries, k=k, by_prob=by_prob, **kwargs,`
`74`	`74`	`)`
`75`	`75`	`if isinstance(passages[0], List):`
`76`	`76`	`pred_returns = []`