update init

basf · Apr 23, 2024 · c9c74cb · c9c74cb
1 parent 880f55c
commit c9c74cb
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/molpipeline/estimators/chemprop/__init__.py b/molpipeline/estimators/chemprop/__init__.py
@@ -4,8 +4,18 @@
 
 installed_packages = {pkg.name for pkg in pkgutil.iter_modules()}
 if "chemprop" in installed_packages:
-    from molpipeline.estimators.chemprop.models import ChempropModel  # noqa
+    from molpipeline.estimators.chemprop.models import (
+        ChempropClassifier,
+        ChempropModel,
+        ChempropNeuralFP,
+        ChempropRegressor,
+    )
 
-    __all__ = ["ChempropModel"]
+    __all__ = [
+        "ChempropClassifier",
+        "ChempropModel",
+        "ChempropNeuralFP",
+        "ChempropRegressor",
+    ]
 else:
     __all__ = []