test

mosaicml · Jun 25, 2024 · 13ecba9 · 13ecba9
1 parent 072758e
commit 13ecba9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/trainer/test_fsdp_checkpoint.py b/tests/trainer/test_fsdp_checkpoint.py
@@ -788,7 +788,7 @@ def mock_get_checkpoint_validation_function():
 
 
 @pytest.mark.gpu
-@pytest.mark.parametrize('use_remote', [pytest.param(True, marks=pytest.mark.remote), False])
+@pytest.mark.parametrize('use_remote', [False])
 @pytest.mark.parametrize(
     'weights_only,optimizer,precision,autoresume,load_ignore_keys,use_symlink,use_tp,use_hsdp',
     [