ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-22 07:15:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix PY2 compatibility issue

Browse Source
This commit is contained in:
allegroai 2021-09-17 16:17:43 +03:00
parent 889d237398
commit 51d70efbff
1 changed files with 28 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
clearml/__init__.py
View File

@ -1,4 +1,5 @@
""" ClearML open SDK """ """ ClearML open SDK """
from six import PY2
from .version import __version__ from .version import __version__
from .task import Task from .task import Task
@ -7,7 +8,31 @@ from .logger import Logger
from .storage import StorageManager from .storage import StorageManager
from .errors import UsageError from .errors import UsageError
from .datasets import Dataset from .datasets import Dataset
if not PY2:
from .automation.controller import PipelineController from .automation.controller import PipelineController
__all__ = ["__version__", "Task", "InputModel", "OutputModel", "Model", "Logger", __all__ = [
"StorageManager", "UsageError", "Dataset", "PipelineController"] "__version__",
"Task",
"InputModel",
"OutputModel",
"Model",
"Logger",
"StorageManager",
"UsageError",
"Dataset",
"PipelineController",
]
else:
__all__ = [
"__version__",
"Task",
"InputModel",
"OutputModel",
"Model",
"Logger",
"StorageManager",
"UsageError",
"Dataset",
]