adjust metrics comparison (#1798)

Signed-off-by: MrPresent-Han <[email protected]>
milvus-io · Nov 22, 2023 · 9075512 · 9075512
1 parent 642f729
commit 9075512
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymilvus/orm/iterator.py b/pymilvus/orm/iterator.py
@@ -203,9 +203,9 @@ def close(self) -> None:
 def metrics_positive_related(metrics: str) -> bool:
     if metrics in [CALC_DIST_L2, CALC_DIST_JACCARD, CALC_DIST_HAMMING, CALC_DIST_TANIMOTO]:
         return True
-    if metrics is CALC_DIST_IP or metrics is CALC_DIST_COSINE:
+    if metrics in [CALC_DIST_IP, CALC_DIST_COSINE]:
         return False
-    raise MilvusException(message=f"unsupported metrics type for search iteration{metrics}")
+    raise MilvusException(message=f"unsupported metrics type for search iteration: {metrics}")
 
 
 class SearchPage(LoopBase):