Removed Trains BC from examples (#3029)

pytorch · Aug 20, 2023 · 02aab6f · 02aab6f
1 parent 932ddd4
commit 02aab6f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 15 deletions.
diff --git a/examples/cifar10/main.py b/examples/cifar10/main.py
@@ -46,11 +46,7 @@ def training(local_rank, config):
             config["cuda device name"] = torch.cuda.get_device_name(local_rank)
 
         if config["with_clearml"]:
-            try:
-                from clearml import Task
-            except ImportError:
-                # Backwards-compatibility for legacy Trains SDK
-                from trains import Task
+            from clearml import Task
 
             task = Task.init("CIFAR10-Training", task_name=output_path.stem)
             task.connect_configuration(config)

diff --git a/examples/cifar10_qat/main.py b/examples/cifar10_qat/main.py
@@ -42,11 +42,7 @@ def training(local_rank, config):
             config["cuda device name"] = torch.cuda.get_device_name(local_rank)
 
         if config["with_clearml"]:
-            try:
-                from clearml import Task
-            except ImportError:
-                # Backwards-compatibility for legacy Trains SDK
-                from trains import Task
+            from clearml import Task
 
             task = Task.init("CIFAR10-Training", task_name=output_path.stem)
             task.connect_configuration(config)

diff --git a/examples/transformers/main.py b/examples/transformers/main.py
@@ -44,11 +44,7 @@ def training(local_rank, config):
             config["cuda device name"] = torch.cuda.get_device_name(local_rank)
 
         if config["with_clearml"]:
-            try:
-                from clearml import Task
-            except ImportError:
-                # Backwards-compatibility for legacy Trains SDK
-                from trains import Task
+            from clearml import Task
 
             task = Task.init("IMDB-Training", task_name=output_path.stem)
             task.connect_configuration(config)