Do not raise an error when a checkpoint path doesn't exist

PiperOrigin-RevId: 731451478
google · Feb 26, 2025 · 78001fa · 78001fa
1 parent c69a872
commit 78001fa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/checkpoint/orbax/checkpoint/_src/path/step.py b/checkpoint/orbax/checkpoint/_src/path/step.py
@@ -523,7 +523,8 @@ def is_tmp_checkpoint(path: epath.PathLike) -> bool:
   """Determines whether a directory is a tmp checkpoint path."""
   path = epath.Path(path)
   if not path.exists():
-    raise ValueError(f'Path {path} does not exist.')
+    logging.warning('Path %s does not exist.', path)
+    return False
   if not path.is_dir():
     return False
   if is_gcs_path(path) and not (path / _COMMIT_SUCCESS_FILE).exists():