diff --git a/src/gretel_trainer/relational/artifacts.py b/src/gretel_trainer/relational/artifacts.py
index e1a835b1..33d1843b 100644
--- 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
index 74be2afa..7f038785 100644
--- 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
index 516dc421..5eb9a70a 100644
--- 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