remove unused imports

delphi-suite · Mar 7, 2024 · 4325012 · 4325012
1 parent bbddba8
commit 4325012
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git 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
@@ -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