commit change

mosaicml · Oct 14, 2024 · 029f5b8 · 029f5b8
1 parent 3f38dc2
commit 029f5b8
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/trainer/test_checkpoint.py b/tests/trainer/test_checkpoint.py
@@ -1715,12 +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(
-                exc_info.value
-            )
+            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)
 
     @pytest.mark.parametrize('spin_dataloaders', [False, True])
     def test_spin_dataloaders(