add remote

mosaicml · Jun 25, 2024 · e3106bf · e3106bf
1 parent c27f1e6
commit e3106bf
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
@@ -793,7 +793,7 @@ def mock_get_checkpoint_validation_function():
 
 
 @pytest.mark.gpu
-@pytest.mark.parametrize('use_remote', [False])
+@pytest.mark.parametrize('use_remote', [pytest.param(True, marks=pytest.mark.remote), False])
 @pytest.mark.parametrize(
     'weights_only,optimizer,precision,autoresume,load_ignore_keys,use_symlink,use_tp,use_hsdp',
     [