Remove warning for scalar divide with inf

agourlay · joein · commit bbe013182ef4 · 2024-04-22T12:27:53.000+02:00
diff --git a/qdrant_client/local/distances.py b/qdrant_client/local/distances.py
@@ -165,10 +165,9 @@ def calculate_distance_core(
 
 
 def fast_sigmoid(x: np.float32) -> np.float32:
-    if np.isnan(x):
-        # To avoid divisions on NaNs, which gets: RuntimeWarning: invalid value encountered in scalar divide
-        return x  # NaN
-
+    if np.isnan(x) or np.isinf(x):
+        # To avoid divisions on NaNs or inf, which gets: RuntimeWarning: invalid value encountered in scalar divide
+        return x
     return x / np.add(1.0, abs(x))
 
 
