added more explicit error handling in colab

HenriquesLab · Jan 23, 2024 · e632cff · e632cff
1 parent f2a0ed3
commit e632cff
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/nanopyx/core/transform/convolution.py b/src/nanopyx/core/transform/convolution.py
@@ -84,6 +84,9 @@ def convolution2D_transonic(image, kernel):
     except ImportError:
         print("Transonic is not installed, defaulting to Python")
         return convolution2D_python(image, kernel)
+    except ModuleNotFoundError:
+        print("Transonic is not installed, defaulting to Python")
+        return convolution2D_python(image, kernel)
 
 
 @njit(cache=True, parallel=True)