fix

allenai · Jan 24, 2025 · 842e927 · 842e927
1 parent 6207bf1
commit 842e927
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/olmo_core/distributed/checkpoint/__init__.py b/src/olmo_core/distributed/checkpoint/__init__.py
@@ -584,8 +584,9 @@ def get_checkpoint_metadata(dir: PathOrStr) -> Metadata:
         return storage_reader.read_metadata()
     except FileNotFoundError as exc:
         msg = f"'{dir}' does not appear to contain a state dict checkpoint."
-        if file_exists((suggested_path := join_path(dir, "model_and_optim/.metadata"))):
-            msg += f" Did you mean to use '{suggested_path}'?"
+        suggested_dir = join_path(dir, "model_and_optim")
+        if file_exists(join_path(suggested_dir, ".metadata")):
+            msg += f" Did you mean to use '{suggested_dir}'?"
         raise FileNotFoundError(msg) from exc