Handle "All-NaN slice encountered" warnings for video.normalize_pixel…

…wise
cardiacvision · Oct 26, 2023 · 66043ef · 66043ef
1 parent 2bec762
commit 66043ef
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/optimap/video/_filters.py b/optimap/video/_filters.py
@@ -1,3 +1,5 @@
+import warnings
+
 import numpy as np
 from scipy import ndimage
 
@@ -66,8 +68,10 @@ def normalize_pixelwise(video: np.ndarray, ymin=0, ymax=1):
     """
     _print(f"normalizing video pixel-wise to interval [{ymin}, {ymax}] ...")
     video = video.astype("float32")
-    min_ = np.nanmin(video, axis=0)
-    max_ = np.nanmax(video, axis=0)
+    with warnings.catch_warnings():  # ignore "All-NaN slice encountered" warnings
+        warnings.simplefilter("ignore", category=RuntimeWarning)
+        min_ = np.nanmin(video, axis=0)
+        max_ = np.nanmax(video, axis=0)
     eps = np.finfo(np.float32).eps
     video = (video - min_) / (max_ - min_ + eps) * (ymax - ymin) + ymin
     return video