mirror of
https://github.com/clearml/clearml
synced 2025-02-12 07:35:08 +00:00
dd3d4cec94
* add tasktypes to main namespace * add tasktypes to pipe decorator example * minor linting Co-authored-by: Johnathan Alexander <jalexander86@gatech.edu>
43 lines
870 B
Python
43 lines
870 B
Python
""" ClearML open SDK """
|
|
from six import PY2
|
|
|
|
from .version import __version__
|
|
from .task import Task
|
|
from .model import InputModel, OutputModel, Model
|
|
from .logger import Logger
|
|
from .storage import StorageManager
|
|
from .errors import UsageError
|
|
from .datasets import Dataset
|
|
|
|
TaskTypes = Task.TaskTypes
|
|
|
|
if not PY2:
|
|
from .automation.controller import PipelineController
|
|
|
|
__all__ = [
|
|
"__version__",
|
|
"Task",
|
|
"TaskTypes",
|
|
"InputModel",
|
|
"OutputModel",
|
|
"Model",
|
|
"Logger",
|
|
"StorageManager",
|
|
"UsageError",
|
|
"Dataset",
|
|
"PipelineController",
|
|
]
|
|
else:
|
|
__all__ = [
|
|
"__version__",
|
|
"Task",
|
|
"TaskTypes",
|
|
"InputModel",
|
|
"OutputModel",
|
|
"Model",
|
|
"Logger",
|
|
"StorageManager",
|
|
"UsageError",
|
|
"Dataset",
|
|
]
|