v0.9.4

batchflow/batch_image.py

@@ -13,8 +13,7 @@
 except ImportError:
     pass
 
-from scipy.ndimage.filters import gaussian_filter
-from scipy.ndimage.interpolation import map_coordinates
+from scipy.ndimage import gaussian_filter, map_coordinates
 
 from .batch import Batch
 from .decorators import action, apply_parallel, inbatch_parallel

batchflow/monitor.py

@@ -318,7 +318,7 @@ class GPUResourceMonitor(ResourceMonitor):
     """ If the `CUDA_VISIBLE_DEVICES` is set, check it and return device numbers. Otherwise, return [0]. """
     def __init__(self, function=None, frequency=0.1, gpu_list=None, **kwargs):
         if pynvml is None:
-            raise ImportError('Install Python interface for pynvml')
+            raise ImportError('Install nvidia-ml-py to use GPU monitoring functionality')
         super().__init__(function=function, frequency=frequency, **kwargs)
 
         # Fallback to env variable

pyproject.toml

@@ -42,7 +42,7 @@ research = [
 
 monitor = [
     "psutil",
-    "pynvml"
+    "nvidia-ml-py"
 ]
 
 nn = [