Silence NumPy warnings about numerical operations with zero.

BiocPy · Dec 20, 2024 · c36c443 · c36c443
1 parent a9f50fe
commit c36c443
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/delayedarray/UnaryIsometricOpSimple.py b/src/delayedarray/UnaryIsometricOpSimple.py
@@ -1,6 +1,7 @@
 from typing import Callable, Literal, Tuple, Sequence
 import numpy
 from numpy import dtype, zeros
+import warnings
 
 from .DelayedOp import DelayedOp
 from .SparseNdarray import SparseNdarray
@@ -68,7 +69,9 @@ def __init__(self, seed, operation: OP):
                 String specifying the unary operation.
         """
         f = _choose_operator(operation)
-        dummy = f(zeros(1, dtype=seed.dtype))
+        with warnings.catch_warnings():  # silence warnings from divide by zero.
+            warnings.simplefilter("ignore")
+            dummy = f(zeros(1, dtype=seed.dtype))
 
         self._seed = seed
         self._op = operation