Skip to content

Commit 2a76769

Browse files
MrPresent-HanMrPresent-Han
andauthored
enhance:expose reduce_stop_for_best to users(#2181) (#2183)
related: #2181 Signed-off-by: MrPresent-Han <[email protected]> Co-authored-by: MrPresent-Han <[email protected]>
1 parent 62434ca commit 2a76769

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pymilvus/orm/iterator.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,10 @@ def __set_up_iteration_states(self):
9292
self._kwargs[ITERATOR_FIELD] = "True"
9393

9494
def __check_set_reduce_stop_for_best(self):
95-
self._kwargs[REDUCE_STOP_FOR_BEST] = "True"
95+
if self._kwargs.get(REDUCE_STOP_FOR_BEST, True):
96+
self._kwargs[REDUCE_STOP_FOR_BEST] = "True"
97+
else:
98+
self._kwargs[REDUCE_STOP_FOR_BEST] = "False"
9699

97100
def __check_set_batch_size(self, batch_size: int):
98101
if batch_size < 0:

0 commit comments

Comments
 (0)