Internal change.

PiperOrigin-RevId: 650662108
google · Jul 9, 2024 · 1e06498 · 1e06498
1 parent a48cc4c
commit 1e06498
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/checkpoint/orbax/checkpoint/type_handlers.py b/checkpoint/orbax/checkpoint/type_handlers.py
@@ -627,8 +627,9 @@ def check_input_arguments(*args):
       raise ValueError('Found input args with mismatched lengths.')
 
 
-def is_ocdbt_checkpoint(path: epath.Path) -> bool:
+def is_ocdbt_checkpoint(path: epath.PathLike) -> bool:
   """Determines whether a checkpoint uses OCDBT format."""
+  path = epath.Path(path)
   return (path / _OCDBT_MANIFEST_FILE).exists()