diff --git a/clearml/__init__.py b/clearml/__init__.py index 0ad97be4..d127f0a5 100644 --- a/clearml/__init__.py +++ b/clearml/__init__.py @@ -7,6 +7,7 @@ from .logger import Logger from .storage import StorageManager from .errors import UsageError from .datasets import Dataset +from .automation.controller import PipelineController __all__ = ["__version__", "Task", "InputModel", "OutputModel", "Model", "Logger", - "StorageManager", "UsageError", "Dataset"] + "StorageManager", "UsageError", "Dataset", "PipelineController"]