Use warnings module

cardiacvision · Oct 26, 2023 · eb64693 · eb64693
1 parent a083fea
commit eb64693
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 7 deletions.
diff --git a/optimap/motion/_flowestimator.py b/optimap/motion/_flowestimator.py
@@ -1,4 +1,5 @@
 from typing import Union
+import warnings
 
 import cv2
 import numpy as np
@@ -129,7 +130,7 @@ def _check_and_convert(imgs: Union[np.ndarray, list], method: str):
         if isinstance(imgs, list):
             imgs = np.array(imgs)
         if imgs.min() < 0 or imgs.max() > 1:
-            print(
+            warnings.warn(
                 "WARNING: image values are not in range [0,1], which may lead to unexpected motion tracking"
             )
         return (imgs * 255).astype(np.uint8)

diff --git a/optimap/utils.py b/optimap/utils.py
@@ -14,6 +14,7 @@
 
 import functools
 import urllib.parse
+import warnings
 
 import matplotlib as mpl
 import matplotlib.pyplot as plt
@@ -168,7 +169,7 @@ def retrieve_example_data(name, directory="./example_data", silent=False):
     """
     known_hash = FILE_HASHES.get(name, None)
     if known_hash is None:
-        print(f"WARNING: Example file '{name}' is not known. Attempting to download it anyway.")
+        warnings.warn(f"WARNING: Example file '{name}' is not known. Attempting to download it anyway.", UserWarning)
     if silent:
         silent = pooch.get_logger().level
         pooch.get_logger().setLevel("WARNING")

diff --git a/optimap/video/_importers.py b/optimap/video/_importers.py
@@ -3,6 +3,7 @@
 from datetime import datetime
 from pathlib import Path
 from typing import BinaryIO
+import warnings
 
 import numpy as np
 
@@ -39,7 +40,7 @@ def load_video(self, start_frame=0, frames=None, step=1, use_mmap=False):
         if frames is not None:
             nframes = frames
             if nframes > self._Nt - start_frame:
-                print(f"Warning: requested {nframes} frames, but only {self._Nt - start_frame} frames available. Loading all available frames.")
+                warnings.warn(f"requested {nframes} frames, but only {self._Nt - start_frame} frames available. Loading all available frames.", UserWarning)
                 nframes = self._Nt - start_frame
         else:
             nframes = self._Nt - start_frame
@@ -182,7 +183,7 @@ def load_video(self, start_frame=0, frames=None, step=1):
         if frames is not None:
             end_frame = start_frame + frames
             if end_frame > self._Nt:
-                print(f"Warning: requested {frames} frames, but only {self._Nt - start_frame} frames available. Loading all available frames.")
+                warnings.warn(f"requested {frames} frames, but only {self._Nt - start_frame} frames available. Loading all available frames.", UserWarning)
                 end_frame = None
         else:
             end_frame = None
@@ -281,7 +282,7 @@ def load_video(self, start_frame=0, frames=None, step=1):
         if frames is not None:
             end_frame = start_frame + frames
             if end_frame > self._Nt:
-                print(f"Warning: requested {frames} frames, but only {self._Nt - start_frame} frames available. Loading all available frames.")
+                warnings.warn(f"requested {frames} frames, but only {self._Nt - start_frame} frames available. Loading all available frames.", UserWarning)
                 end_frame = None
         else:
             end_frame = None

diff --git a/optimap/video/_load.py b/optimap/video/_load.py
@@ -1,5 +1,6 @@
 import re
 from pathlib import Path
+import warnings
 
 import numpy as np
 import skimage.io as sio
@@ -110,8 +111,7 @@ def load_MATLAB(filename, fieldname=None, start_frame=0, end_frame=None, step=1)
         if len(fields) == 0:
             raise ValueError(f"No fields found in file '{filename}'")
         elif len(fields) > 1:
-            print(f"WARNING: Multiple fields found in file '{filename}', loading first field '{fields[0]}'"
-            )
+            warnings.warn(f"Multiple fields found in file '{filename}', loading first field '{fields[0]}'", UserWarning)
         fieldname = fields[0]
     video = data[fieldname]
     return video[start_frame:end_frame:step]