Added cupy version check in the arrus.utils.imaging package. (#243)

* cupy version check added to imaging.py * Minor update. Co-authored-by: Piotr Jarosik <piotr.jarosik@us4us.eu>
us4useu · Nov 5, 2021 · 44e55eb · 44e55eb
1 parent 3158570
commit 44e55eb
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/api/python/arrus/utils/imaging.py b/api/python/arrus/utils/imaging.py
@@ -15,6 +15,11 @@
 from collections import deque
 from collections.abc import Iterable
 
+import cupy
+if cupy.__version__ < "9.0.0":
+    raise Exception(f"The version of cupy module is too low. "
+                    f"Try install the version ''9.0.0'' or higher.")
+
 
 def get_extent(x_grid, z_grid):
     """