fix: NoneType epochs

2025-12-10 15:37:54 +01:00 · 2025-12-10 15:37:54 +01:00 · 6c5908e6ae
commit 6c5908e6ae
parent 63119980c9
4 changed files with 6 additions and 4 deletions
--- a/src/trainers/trainer.py
+++ b/src/trainers/trainer.py
@ -13,7 +13,7 @@ class Trainer(ABC):
            model: nn.Module | None,
            train_loader: DataLoader,
            validation_loader: DataLoader,
-            n_epochs: int,
+            n_epochs: int | None,
            device: str
    ) -> nn.Module:
        pass