Fix typo in error message when unexpected best_mode passed in to ch…

…eckpoint manager. PiperOrigin-RevId: 713812053
google · Jan 9, 2025 · 21b9939 · 21b9939
1 parent 9899e97
commit 21b9939
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/checkpoint/orbax/checkpoint/checkpoint_manager.py b/checkpoint/orbax/checkpoint/checkpoint_manager.py
@@ -298,7 +298,7 @@ def __post_init__(self):
     if self.best_mode not in ('min', 'max'):
       msg = (
           "`CheckpointManagerOptions.best_mode` must be one of None, 'min' "
-          "or 'max'. Got {self.dtype}."
+          f"or 'max'. Got {self.best_mode}."
       )
       raise ValueError(msg)
     if self.max_to_keep is not None and self.max_to_keep < 0: