Remove default value for batches_early_stop parameter (#150)

Deci-AI · Jul 10, 2023 · 5ea3f0b · 5ea3f0b
1 parent 55bbfac
commit 5ea3f0b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/data_gradients/managers/detection_manager.py b/src/data_gradients/managers/detection_manager.py
@@ -33,7 +33,7 @@ def __init__(
         is_label_first: Optional[bool] = None,
         bbox_format: Optional[str] = None,
         n_image_channels: int = 3,
-        batches_early_stop: int = 999,
+        batches_early_stop: Optional[int] = None,
         remove_plots_after_report: Optional[bool] = True,
     ):
         """
@@ -108,5 +108,5 @@ def __init__(
             grouped_feature_extractors=grouped_feature_extractors,
             log_dir=log_dir,
             batches_early_stop=batches_early_stop,
-            remove_plots_after_report=remove_plots_after_report
+            remove_plots_after_report=remove_plots_after_report,
         )
diff --git a/src/data_gradients/managers/segmentation_manager.py b/src/data_gradients/managers/segmentation_manager.py
@@ -32,7 +32,7 @@ def __init__(
         labels_extractor: Optional[Callable[[SupportedDataType], torch.Tensor]] = None,
         num_image_channels: int = 3,
         threshold_soft_labels: float = 0.5,
-        batches_early_stop: int = 999,
+        batches_early_stop: Optional[int] = None,
         remove_plots_after_report: Optional[bool] = True,
     ):
         """