Merge pull request #186 from DiamondLightSource/fixdenimp

fixes the import for denoisers
DiamondLightSource · Jan 17, 2025 · d7090ec · d7090ec
2 parents df49be2 + e472bb3
commit d7090ec
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/httomolibgpu/misc/denoise.py b/httomolibgpu/misc/denoise.py
@@ -32,7 +32,12 @@
 from numpy import float32
 from unittest.mock import Mock
 
-from ccpi.filters.regularisersCuPy import ROF_TV, PD_TV
+if cupy_run:
+    from ccpi.filters.regularisersCuPy import ROF_TV, PD_TV
+else:
+    ROF_TV = Mock()
+    PD_TV = Mock()
+
 
 __all__ = [
     "total_variation_ROF",