enhance:expose reduce_stop_for_best to users(milvus-io#2181)

Signed-off-by: MrPresent-Han <[email protected]>
MrPresent-Han · Jul 19, 2024 · 7c7b92b · 7c7b92b
1 parent 1b0215c
commit 7c7b92b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pymilvus/orm/iterator.py b/pymilvus/orm/iterator.py
@@ -92,7 +92,10 @@ def __set_up_iteration_states(self):
         self._kwargs[ITERATOR_FIELD] = "True"
 
     def __check_set_reduce_stop_for_best(self):
-        self._kwargs[REDUCE_STOP_FOR_BEST] = "True"
+        if self._kwargs.get(REDUCE_STOP_FOR_BEST, True):
+            self._kwargs[REDUCE_STOP_FOR_BEST] = "True"
+        else:
+            self._kwargs[REDUCE_STOP_FOR_BEST] = "False"
 
     def __check_set_batch_size(self, batch_size: int):
         if batch_size < 0: