cleanlab · jwmueller · Jun 24, 2023 · Oct 11, 2022 · Nov 4, 2022 · Nov 10, 2022
diff --git a/cleanlab/__init__.py b/cleanlab/__init__.py
@@ -10,3 +10,46 @@
 from . import token_classification
 from . import multilabel_classification
 from . import regression
+
+
+class DatalabUnavailable:
+    def __init__(self, message):
+        self.message = message
+
+    def __getattr__(self, name):
+        message = self.message + f" (raised when trying to access {name})"
+        raise ImportError(message)
+
+    def __call__(self, *args, **kwargs):
+        message = (
+            self.message + f" (raised when trying to call with args: {args}, kwargs: {kwargs})"
+        )
+        raise ImportError(message)
+
+
+def _datalab_import_factory():
+    try:
+        from .datalab.datalab import Datalab as _Datalab
+
+        return _Datalab
+    except ImportError:
+        return DatalabUnavailable(
+            "Datalab is not available due to missing dependencies. "
+            "To install Datalab, run `pip install cleanlab[datalab]`."
+        )
+
+
+def _issue_manager_import_factory():
+    try:
+        from .datalab.issue_manager import IssueManager as _IssueManager
+
+        return _IssueManager
+    except ImportError:
+        return DatalabUnavailable(
+            "IssueManager is not available due to missing dependencies for Datalab. "
+            "To install Datalab, run `pip install cleanlab[datalab]`."
+        )
+
+
+Datalab = _datalab_import_factory()
+IssueManager = _issue_manager_import_factory()
diff --git a/setup.cfg b/setup.cfg
@@ -6,3 +6,6 @@ per-file-ignores =
     cleanlab/token_classification/__init__.py: F401
     cleanlab/benchmarking/__init__.py: F401
     cleanlab/regression/__init__.py: F401
+    cleanlab/datalab/*/__init__.py: F401
+    cleanlab/models/__init__.py: F401
+    cleanlab/multilabel_classification/__init__.py: F401