add imports module

anas-rz · Mar 1, 2024 · 560f580 · 560f580
1 parent f900919
commit 560f580
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/k3_addons/utils/imports.py b/k3_addons/utils/imports.py
@@ -0,0 +1,22 @@
+# create a variable and check if pysoti is available and return a constant PYSOTI_AVAILABLE
+
+try:
+    import pystoi
+
+    PYSOTI_AVAILABLE = True
+except ImportError:
+    PYSOTI_AVAILABLE = False
+
+try:
+    import pesq
+
+    PESQ_AVAILABLE = True
+except ImportError:
+    PESQ_AVAILABLE = False
+
+try:
+    import multiprocessing
+
+    MULTIPROCESSING_AVAILABLE = True
+except ImportError:
+    MULTIPROCESSING_AVAILABLE = False