enhance:expose reduce_stop_for_best to users(#2181) (#2188)

related: #2181 Signed-off-by: MrPresent-Han <[email protected]> Co-authored-by: MrPresent-Han <[email protected]>
milvus-io · Jul 22, 2024 · 93be83a · 93be83a
1 parent cc1b448
commit 93be83a
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: