is_package_lite demo

py4dstem · Jul 10, 2024 · 1e8278f · 1e8278f
1 parent 7b60b13
commit 1e8278f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/py4DSTEM/__init__.py b/py4DSTEM/__init__.py
@@ -1,6 +1,9 @@
 from py4DSTEM.version import __version__
 from emdfile import tqdmnd
 
+from importlib.metadata import metadata
+
+is_package_lite = "lite" in metadata("py4DSTEM")["Keywords"].lower().split(",")
 
 ### io
 
@@ -54,9 +57,9 @@
 
 try:
     from py4DSTEM.process import classification
-except (ImportError, ModuleNotFoundError):
-    pass
-
+except (ImportError, ModuleNotFoundError) as exc:
+    if not is_package_lite:
+        raise exc
 
 # diffraction
 from py4DSTEM.process.diffraction import Crystal, Orientation
@@ -75,8 +78,9 @@
 
 try:
     from py4DSTEM.process import wholepatternfit
-except (ImportError, ModuleNotFoundError):
-    pass
+except (ImportError, ModuleNotFoundError) as exc:
+    if not is_package_lite:
+        raise exc
 
 
 ### more submodules

diff --git a/setup.py b/setup.py
@@ -20,7 +20,7 @@
     author="Benjamin H. Savitzky",
     author_email="[email protected]",
     license="GNU GPLv3",
-    keywords="STEM 4DSTEM",
+    keywords="STEM,4DSTEM",
     python_requires=">=3.10",
     install_requires=[
         "numpy >= 1.19",