From 43250128d5c6e3e2dac0a8bc7faa3e4193b5132a Mon Sep 17 00:00:00 2001
From: JaiDhyani <jaiwithani@gmail.com>
Date: Thu, 7 Mar 2024 08:22:59 -0800
Subject: [PATCH] remove unused imports

---
 src/delphi/train/architectures.py | 1 -
 src/delphi/train/utils.py         | 7 +------
 2 files changed, 1 insertion(+), 7 deletions(-)

diff --git a/src/delphi/train/architectures.py b/src/delphi/train/architectures.py
index 83453151..693fbc70 100644
--- a/src/delphi/train/architectures.py
+++ b/src/delphi/train/architectures.py
@@ -1,4 +1,3 @@
-import os
 from dataclasses import fields
 
 import torch
diff --git a/src/delphi/train/utils.py b/src/delphi/train/utils.py
index a7381a99..f501ceeb 100644
--- a/src/delphi/train/utils.py
+++ b/src/delphi/train/utils.py
@@ -12,12 +12,7 @@
 
 from delphi import constants
 from delphi.eval.utils import load_delphi_dataset
-from delphi.train.architectures import (
-    ModelTypes,
-    export_model,
-    initialize_model,
-    load_model,
-)
+from delphi.train.architectures import export_model, initialize_model, load_model
 from delphi.train.gigaconfig import GigaConfig
 from delphi.train.tokenized_chunks_dataset import TokenizedChunksDataset