Internal change

PiperOrigin-RevId: 944615290

Author: RecML authors

recml/core/training/keras_trainer.py

@@ -122,7 +122,7 @@ def __init__(
       max_checkpoints_to_keep: int = 5,
       checkpoint_save_interval_epochs: int = 1,
       rng_seed: int = core.DEFAULT_RNG_SEED,
-      legacy_checkpoint_format: bool = True,
+      legacy_checkpoint_format: bool = False,
   ):
     """Initializes the instance."""
 
@@ -166,7 +166,7 @@ def train_callbacks(self) -> list[keras.callbacks.Callback]:
             save_interval_epochs=self._checkpoint_save_interval_epochs,
         )
       else:
-        checkpoint_manager = keras_utils.KerasOrbaxCheckpointManagerV2(
+        checkpoint_manager = keras_utils.KerasOrbaxCheckpointManagerV3(
             checkpoint_dir=self._checkpoint_dir,
             max_to_keep=self._max_checkpoints_to_keep,
             save_interval_epochs=self._checkpoint_save_interval_epochs,