Some import checks

openPMD · Feb 4, 2025 · 8feaf18 · 8feaf18
1 parent dc2af3f
commit 8feaf18
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 4 deletions.
diff --git a/examples/15_scipp_loader.py b/examples/15_scipp_loader.py
@@ -1,13 +1,19 @@
 import openpmd_api as pmd
-import openpmd_api.scipp as pmdsc
-import scipp as sc
 
 
 def main():
     series = pmd.Series("../samples/git-sample/data%T.h5", pmd.Access.read_only)
 
+    try:
+        scipp_loader = series.to_scipp()
+        import plopp
+    except ImportError:
+        print("Need to install scipp and plopp to run this example.")
+        return
+    import openpmd_api.scipp as pmdsc
+    import scipp as sc
     time = 65 * sc.Unit("fs")
-    scipp_loader = series.to_scipp()
+
     print(scipp_loader.iterations)
     Ex = scipp_loader.get_field("E", "x", time=time)
     print(Ex)

diff --git a/src/binding/python/openpmd_api/ScippLazyInit.py b/src/binding/python/openpmd_api/ScippLazyInit.py
@@ -1,6 +1,15 @@
 def series_to_scipp(series):
 
-    import scipp
+    # lazy import
+    try:
+        import scipp
+        found_scipp = True
+    except ImportError as original_error:
+        found_scipp = False
+        original_error_string = f"{original_error}"
+
+    if not found_scipp:
+        raise ImportError(f"Scipp NOT found. Install scipp for Scipp support. Original error: {original_error_string}")
 
     from .scipp import DataLoader