From 66043efeefe1e7a530770d42ccb60f91ca267377 Mon Sep 17 00:00:00 2001 From: Jan Lebert Date: Wed, 25 Oct 2023 21:26:46 -0700 Subject: [PATCH] Handle "All-NaN slice encountered" warnings for video.normalize_pixelwise --- optimap/video/_filters.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/optimap/video/_filters.py b/optimap/video/_filters.py index b43079a..fb62118 100644 --- 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