add missing default value for early stopping

mu_map/training/cgan.py

@@ -84,7 +84,7 @@ class cGANTraining(AbstractTraining):
         loss_func_dist: WeightedLoss,
         weight_criterion_dist: float,
         weight_criterion_adv: float,
-        early_stopping: Optional[int],
+        early_stopping: Optional[int] = None,
         logger: Optional[Logger] = None,
     ):
         super().__init__(

mu_map/training/lib.py

@@ -70,7 +70,7 @@ class AbstractTraining:
         device: torch.device,
         snapshot_dir: str,
         snapshot_epoch: int,
-        early_stopping: Optional[int],
+        early_stopping: Optional[int] = None,
         logger: Optional[Logger],
     ):
         self.epochs = epochs