fix: instantiate concrete FakeModel in tests instead of TrainedModel

spirabr · Sep 7, 2024 · b2093fb · b2093fb
1 parent a17e74d
commit b2093fb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/fakes/fake_trained_models_repository.py b/tests/fakes/fake_trained_models_repository.py
@@ -3,6 +3,7 @@
 from src.spira_training.shared.ports.trained_models_repository import (
     TrainedModelsRepository,
 )
+from tests.fakes.fake_model import FakeModel
 
 
 class FakeTrainedModelsRepository(TrainedModelsRepository):
@@ -18,4 +19,4 @@ async def save_model(self, model: TrainedModel, path: Path) -> None:
 
 
 def make_trained_model():
-    return TrainedModel()
+    return FakeModel()