From 497ba92bff93deb031e705776d9d50938ea51db0 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Wed, 17 Nov 2021 19:13:18 +0200
Subject: [PATCH] Add PipelineDecorator to clearml and clearml.automation
 namespaces

---
 clearml/__init__.py            | 3 ++-
 clearml/automation/__init__.py | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/clearml/__init__.py b/clearml/__init__.py
index 631b2da9..58b9bb72 100644
--- a/clearml/__init__.py
+++ b/clearml/__init__.py
@@ -12,7 +12,7 @@ from .datasets import Dataset
 TaskTypes = Task.TaskTypes
 
 if not PY2:
-    from .automation.controller import PipelineController  # noqa: F401
+    from .automation.controller import PipelineController, PipelineDecorator  # noqa: F401
 
     __all__ = [
         "__version__",
@@ -26,6 +26,7 @@ if not PY2:
         "UsageError",
         "Dataset",
         "PipelineController",
+        "PipelineDecorator",
     ]
 else:
     __all__ = [
diff --git a/clearml/automation/__init__.py b/clearml/automation/__init__.py
index fe0d270f..d77b929b 100644
--- a/clearml/automation/__init__.py
+++ b/clearml/automation/__init__.py
@@ -2,10 +2,10 @@ from .parameters import (UniformParameterRange, DiscreteParameterRange, UniformI
                          LogUniformParameterRange)
 from .optimization import GridSearch, RandomSearch, HyperParameterOptimizer, Objective
 from .job import ClearmlJob
-from .controller import PipelineController
+from .controller import PipelineController, PipelineDecorator
 from .scheduler import TaskScheduler
 from .trigger import TriggerScheduler
 
 __all__ = ["UniformParameterRange", "DiscreteParameterRange", "UniformIntegerParameterRange", "ParameterSet",
            "LogUniformParameterRange", "GridSearch", "RandomSearch", "HyperParameterOptimizer", "Objective",
-           "ClearmlJob", "PipelineController", "TaskScheduler", "TriggerScheduler"]
+           "ClearmlJob", "PipelineController", "TaskScheduler", "TriggerScheduler", "PipelineDecorator"]