fix normalizing dir

allenai · Apr 5, 2024 · 32afdaa · 32afdaa
1 parent b97bb4a
commit 32afdaa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/olmo_core/distributed/checkpoint.py b/src/olmo_core/distributed/checkpoint.py
@@ -627,7 +627,7 @@ def _get_global_save_plan_and_metadata(
         )
 
     def _normalize_dir(self, dir: PathOrStr) -> str:
-        dir = str(dir).strip("/")
+        dir = str(dir).rstrip("/")
         if dir.startswith("file://"):
             dir = dir.replace("file://", "", 1)
         return dir