ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 00:56:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add PipelineDecorator to clearml and clearml.automation namespaces

Browse Source
This commit is contained in:
allegroai 2021-11-17 19:13:18 +02:00
parent 3f7d271c10
commit 497ba92bff
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
clearml/__init__.py
View File

@ -12,7 +12,7 @@ from .datasets import Dataset
TaskTypes = Task.TaskTypes TaskTypes = Task.TaskTypes
if not PY2: if not PY2:
from .automation.controller import PipelineController # noqa: F401 from .automation.controller import PipelineController, PipelineDecorator # noqa: F401
__all__ = [ __all__ = [
"__version__", "__version__",
@ -26,6 +26,7 @@ if not PY2:
"UsageError", "UsageError",
"Dataset", "Dataset",
"PipelineController", "PipelineController",
"PipelineDecorator",
] ]
else: else:
__all__ = [ __all__ = [

4
clearml/automation/__init__.py
View File

@ -2,10 +2,10 @@ from .parameters import (UniformParameterRange, DiscreteParameterRange, UniformI
LogUniformParameterRange) LogUniformParameterRange)
from .optimization import GridSearch, RandomSearch, HyperParameterOptimizer, Objective from .optimization import GridSearch, RandomSearch, HyperParameterOptimizer, Objective
from .job import ClearmlJob from .job import ClearmlJob
from .controller import PipelineController from .controller import PipelineController, PipelineDecorator
from .scheduler import TaskScheduler from .scheduler import TaskScheduler
from .trigger import TriggerScheduler from .trigger import TriggerScheduler
__all__ = ["UniformParameterRange", "DiscreteParameterRange", "UniformIntegerParameterRange", "ParameterSet", __all__ = ["UniformParameterRange", "DiscreteParameterRange", "UniformIntegerParameterRange", "ParameterSet",
"LogUniformParameterRange", "GridSearch", "RandomSearch", "HyperParameterOptimizer", "Objective", "LogUniformParameterRange", "GridSearch", "RandomSearch", "HyperParameterOptimizer", "Objective",
"ClearmlJob", "PipelineController", "TaskScheduler", "TriggerScheduler"] "ClearmlJob", "PipelineController", "TaskScheduler", "TriggerScheduler", "PipelineDecorator"]