commit change

mosaicml · Oct 14, 2024 · edaf21c · edaf21c
1 parent 3f38dc2
commit edaf21c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/trainer/test_checkpoint.py b/tests/trainer/test_checkpoint.py
@@ -1715,10 +1715,9 @@ def test_load_incorrect_path(self, world_size: int, tmp_path: pathlib.Path, capl
         if dist.get_global_rank() == 0:
             assert f'Local path {incorrect_path} does not exist' in str(exc_info.value)
         else:
-            assert f'No such file or directory: {incorrect_path}. This likely implies a download failed on local rank 0 or global rank 0' in str(
-                exc_info.value
-            )
-            assert f'Please check the logs for global rank 0 to debug the checkpoint download issue.' in str(
+            assert 'No such file or directory:' in str(exc_info.value)
+            assert 'This likely implies a download failed on local rank 0 or global rank 0' in str(exc_info.value)
+            assert 'Please check the logs for global rank 0 to debug the checkpoint download issue.' in str(
                 exc_info.value
             )