Skip to content

Commit 0c03487

Browse files
author
SeanNaren
committed
Special the test to see if this fixes CI
1 parent d782554 commit 0c03487

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/plugins/test_deepspeed_plugin.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -181,8 +181,7 @@ def test_deepspeed_defaults(tmpdir):
181181
assert isinstance(plugin.config["zero_optimization"], dict)
182182

183183

184-
@RunIf(deepspeed=True)
185-
@mock.patch('torch.distributed.is_initialized', return_value=True)
184+
@RunIf(min_gpus=1, deepspeed=True, special=True)
186185
def test_invalid_deepspeed_defaults_no_precision(dist_initialized_mock, tmpdir):
187186
"""Test to ensure that using defaults, if precision is not set to 16, we throw an exception."""
188187
model = BoringModel()

0 commit comments

Comments
 (0)