wehub-resource-sync
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from .version import __version__
|
|
from . import classification
|
|
from . import count
|
|
from . import rank
|
|
from . import filter
|
|
from . import benchmarking
|
|
from . import dataset
|
|
from . import multiannotator
|
|
from . import outlier
|
|
from . import token_classification
|
|
from . import multilabel_classification
|
|
from . import object_detection
|
|
from . import regression
|
|
from . import segmentation
|
|
|
|
|
|
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.internal.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()
|