Use `dataclass_from_init` to implement `EarlyStopConfig`.

declearn/main/utils/__init__.py

@@ -6,4 +6,4 @@ from ._checkpoint import Checkpointer
 from ._constraints import Constraint, ConstraintSet, TimeoutConstraint
 from ._data_info import AggregationError, aggregate_clients_data_info
 from ._dataclasses import EvaluateConfig, RegisterConfig, TrainingConfig
-from ._early_stop import EarlyStopping
+from ._early_stop import EarlyStopping, EarlyStopConfig

declearn/main/utils/_early_stop.py

@@ -5,8 +5,12 @@
 from typing import Optional
 
 
+from declearn.utils import dataclass_from_init
+
+
 __all__ = [
     "EarlyStopping",
+    "EarlyStopConfig",
 ]
 
 
@@ -91,3 +95,6 @@ class EarlyStopping:
         else:
             self._n_iter_stuck = 0
         return self.keep_training
+
+
+EarlyStopConfig = dataclass_from_init(EarlyStopping, name="EarlyStopConfig")