using try except -> pkgutil does not work with notebook

basf · Apr 25, 2024 · 11cdd8e · 11cdd8e
1 parent de13674
commit 11cdd8e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/molpipeline/estimators/chemprop/__init__.py b/molpipeline/estimators/chemprop/__init__.py
@@ -1,9 +1,6 @@
 """Initialize Chemprop module."""
 
-import pkgutil
-
-installed_packages = {pkg.name for pkg in pkgutil.iter_modules()}
-if "chemprop" in installed_packages:
+try:
     from molpipeline.estimators.chemprop.models import (  # noqa: F401
         ChempropClassifier,
         ChempropModel,
@@ -17,5 +14,5 @@
         "ChempropNeuralFP",
         "ChempropRegressor",
     ]
-else:
+except ImportError:
     __all__ = []
diff --git a/molpipeline/mol2any/__init__.py b/molpipeline/mol2any/__init__.py
@@ -1,7 +1,5 @@
 """Init the module for mol2any pipeline elements."""
 
-import pkgutil
-
 from molpipeline.mol2any.mol2bin import MolToBinary
 from molpipeline.mol2any.mol2concatinated_vector import MolToConcatenatedVector
 from molpipeline.mol2any.mol2inchi import MolToInchi, MolToInchiKey
@@ -21,8 +19,9 @@
     "MolToRDKitPhysChem",
 ]
 
-installed_packages = {pkg.name for pkg in pkgutil.iter_modules()}
-if "chemprop" in installed_packages:
+try:
     from molpipeline.mol2any.mol2chemprop import MolToChemprop  # noqa
 
     __all__.append("MolToChemprop")
+except ImportError:
+    pass