Remove warning for scalar divide with inf

qdrant · Apr 22, 2024 · bbe0131 · bbe0131
1 parent ef8e772
commit bbe0131
Showing 1 changed file with 3 additions and 4 deletions.
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))