From 51d70efbffa87aa41b46c2024918bf4c584f29cf Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Fri, 17 Sep 2021 16:17:43 +0300
Subject: [PATCH] Fix PY2 compatibility issue

---
 clearml/__init__.py | 31 ++++++++++++++++++++++++++++---
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/clearml/__init__.py b/clearml/__init__.py
index d127f0a5..29dde4ac 100644
--- a/clearml/__init__.py
+++ b/clearml/__init__.py
@@ -1,4 +1,5 @@
 """ ClearML open SDK """
+from six import PY2
 
 from .version import __version__
 from .task import Task
@@ -7,7 +8,31 @@ 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", "PipelineController"]
+if not PY2:
+    from .automation.controller import PipelineController
+
+    __all__ = [
+        "__version__",
+        "Task",
+        "InputModel",
+        "OutputModel",
+        "Model",
+        "Logger",
+        "StorageManager",
+        "UsageError",
+        "Dataset",
+        "PipelineController",
+    ]
+else:
+    __all__ = [
+        "__version__",
+        "Task",
+        "InputModel",
+        "OutputModel",
+        "Model",
+        "Logger",
+        "StorageManager",
+        "UsageError",
+        "Dataset",
+    ]