Clean up some imports (#141)

gretelai · Jul 27, 2023 · 6bc41cc · 6bc41cc
1 parent 795ee98
commit 6bc41cc
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/src/gretel_trainer/relational/artifacts.py b/src/gretel_trainer/relational/artifacts.py
@@ -1,6 +1,4 @@
-import os
 import shutil
-import tarfile
 import tempfile
 from dataclasses import dataclass
 from pathlib import Path

diff --git a/src/gretel_trainer/relational/task_runner.py b/src/gretel_trainer/relational/task_runner.py
@@ -1,9 +1,9 @@
 import logging
 from collections import defaultdict
+from typing import Protocol
 
 from gretel_client.projects.jobs import END_STATES, Job, Status
 from gretel_client.projects.projects import Project
-from typing_extensions import Protocol
 
 from gretel_trainer.relational.sdk_extras import ExtendedGretelSDK
 

diff --git a/src/gretel_trainer/relational/tasks/common.py b/src/gretel_trainer/relational/tasks/common.py
@@ -1,10 +1,9 @@
 import logging
 import time
-from typing import Union
+from typing import Protocol, Union
 
 from gretel_client.projects.jobs import Job, Status
 from gretel_client.projects.projects import Project
-from typing_extensions import Protocol
 
 from gretel_trainer.relational.core import RelationalData
 from gretel_trainer.relational.sdk_extras import ExtendedGretelSDK