diff --git a/src/allencell_ml_segmenter/_tests/training/test_model_selection_widget.py b/src/allencell_ml_segmenter/_tests/training/test_model_selection_widget.py
index 7cbac28b..72b1fe87 100644
--- a/src/allencell_ml_segmenter/_tests/training/test_model_selection_widget.py
+++ b/src/allencell_ml_segmenter/_tests/training/test_model_selection_widget.py
@@ -12,9 +12,6 @@
 from allencell_ml_segmenter.training.training_model import (
     TrainingModel,
 )
-from allencell_ml_segmenter._tests.fakes.fake_experiments_model import (
-    FakeExperimentsModel,
-)
 
 
 @pytest.fixture
diff --git a/src/allencell_ml_segmenter/_tests/training/test_view.py b/src/allencell_ml_segmenter/_tests/training/test_view.py
index 3bf235e1..96c80b24 100644
--- a/src/allencell_ml_segmenter/_tests/training/test_view.py
+++ b/src/allencell_ml_segmenter/_tests/training/test_view.py
@@ -11,6 +11,7 @@
 import pytest
 from pytestqt.qtbot import QtBot
 
+
 @pytest.fixture
 def main_model():
     return MainModel()
@@ -34,7 +35,9 @@ def viewer():
 
 
 @pytest.fixture
-def training_view(qtbot: QtBot, main_model: MainModel, training_model: TrainingModel) -> TrainingView:
+def training_view(
+    qtbot: QtBot, main_model: MainModel, training_model: TrainingModel
+) -> TrainingView:
     """
     Returns a PredictionView instance for testing.
     """
@@ -48,8 +51,7 @@ def training_view(qtbot: QtBot, main_model: MainModel, training_model: TrainingM
 
 
 def test_set_patch_size(
-    training_view: TrainingView,
-    training_model: TrainingModel
+    training_view: TrainingView, training_model: TrainingModel
 ) -> None:
     """
     Tests that using the associated combo box properly sets the patch size field.
diff --git a/src/allencell_ml_segmenter/training/view.py b/src/allencell_ml_segmenter/training/view.py
index 99b319a7..f935ae5d 100644
--- a/src/allencell_ml_segmenter/training/view.py
+++ b/src/allencell_ml_segmenter/training/view.py
@@ -21,7 +21,6 @@
 from allencell_ml_segmenter.core.view import View
 from allencell_ml_segmenter.main.experiments_model import ExperimentsModel
 from allencell_ml_segmenter.main.main_model import MainModel
-from allencell_ml_segmenter.services.training_service import TrainingService
 from allencell_ml_segmenter.training.image_selection_widget import (
     ImageSelectionWidget,
 )