From c26c7971edbfbe72dc94a2123cdeb68ee7ab743e Mon Sep 17 00:00:00 2001 From: Mike Knepper Date: Thu, 27 Jul 2023 14:54:48 -0500 Subject: [PATCH] Clean up some imports --- src/gretel_trainer/relational/artifacts.py | 2 -- src/gretel_trainer/relational/task_runner.py | 2 +- src/gretel_trainer/relational/tasks/common.py | 3 +-- 3 files changed, 2 insertions(+), 5 deletions(-) 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