Update dspy/retrieve/clarifai_rm.py

mogith-pn · sainivedh · web-flow · commit 45797c44d02c · 2024-01-24T17:47:21.000+05:30
Co-authored-by: Sai Nivedh &lt;sainivedh123@gmail.com&gt;
diff --git a/dspy/retrieve/clarifai_rm.py b/dspy/retrieve/clarifai_rm.py
@@ -81,7 +81,7 @@ def forward(
             if isinstance(query_or_queries, str)
             else query_or_queries
         )
-
+self.clarifai_search.top_k = k if k is not None else self.clarifai_search.top_k
         passages = []
         queries = [q for q in queries if q]
 

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ def forward(`
`81`	`81`	`if isinstance(query_or_queries, str)`
`82`	`82`	`else query_or_queries`
`83`	`83`	`)`
`84`		`-`
	`84`	`+self.clarifai_search.top_k = k if k is not None else self.clarifai_search.top_k`
`85`	`85`	`passages = []`
`86`	`86`	`queries = [q for q in queries if q]`
`87`	`87`