Skip to content

Commit d7090ec

Browse files
authored
Merge pull request #186 from DiamondLightSource/fixdenimp
fixes the import for denoisers
2 parents df49be2 + e472bb3 commit d7090ec

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

httomolibgpu/misc/denoise.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,12 @@
3232
from numpy import float32
3333
from unittest.mock import Mock
3434

35-
from ccpi.filters.regularisersCuPy import ROF_TV, PD_TV
35+
if cupy_run:
36+
from ccpi.filters.regularisersCuPy import ROF_TV, PD_TV
37+
else:
38+
ROF_TV = Mock()
39+
PD_TV = Mock()
40+
3641

3742
__all__ = [
3843
"total_variation_ROF",

0 commit comments

Comments
 (0)