add try/except ModuleNotFoundError

CEA-MetroCarac · Sep 3, 2024 · b8ac0b9 · b8ac0b9
1 parent 34c4c21
commit b8ac0b9
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 9 deletions.
diff --git a/examples/ex_real_stack.py b/examples/ex_real_stack.py
@@ -10,8 +10,12 @@
 from pystack3d import Stack3d
 from pystack3d.utils_metadata_fibics import params_from_metadata
 
-from utils import init_dir, postpro, plot_results
-from utils import UserTempDirectory  # pylint: disable=unused-import
+try:
+    from utils import init_dir, postpro, plot_results
+    from utils import UserTempDirectory  # pylint: disable=unused-import
+except ModuleNotFoundError:
+    from examples.utils import init_dir, postpro, plot_results
+    from examples.utils import UserTempDirectory
 
 ASSETS = Path(__file__).parents[1] / 'assets'
 DATA = ASSETS / 'stacks' / 'stack_1'

diff --git a/examples/ex_real_stack_perf.py b/examples/ex_real_stack_perf.py
@@ -8,8 +8,12 @@
 
 from pystack3d import Stack3d
 
-from utils import init_dir
-from utils import UserTempDirectory  # pylint: disable=unused-import
+try:
+    from utils import init_dir
+    from utils import UserTempDirectory  # pylint: disable=unused-import
+except ModuleNotFoundError:
+    from examples.utils import init_dir
+    from examples.utils import UserTempDirectory
 
 ASSETS = Path(__file__).parents[1] / 'assets'
 DATA = ASSETS / 'stacks' / 'stack_1'

diff --git a/examples/ex_synthetic_stack.py b/examples/ex_synthetic_stack.py
@@ -3,20 +3,22 @@
 """
 import os
 from pathlib import Path
-from tempfile import TemporaryDirectory
+from tempfile import TemporaryDirectory  # pylint: disable=unused-import
 import numpy as np
-import matplotlib as mpl
 import matplotlib.pyplot as plt
 from scipy.ndimage import shift
 from tifffile import imwrite, imread
-from skimage.draw import line
 from pyvsnr.utils import curtains_addition
 
 import pystack3d
 from pystack3d import Stack3d
 
-from examples.utils import init_dir, postpro, plot_results, plot_cube_faces
-from examples.utils import UserTempDirectory
+try:
+    from utils import init_dir, postpro, plot_results, plot_cube_faces
+    from utils import UserTempDirectory  # pylint: disable=unused-import
+except ModuleNotFoundError:
+    from examples.utils import init_dir, postpro, plot_results, plot_cube_faces
+    from examples.utils import UserTempDirectory
 
 POLICY = "slice_{slice_nb}_z={z_coord}um.tif"