Skip to content

Commit 088dbbe

Browse files
committed
fix deepspeed compat test after strategy flag change
1 parent 19e055c commit 088dbbe

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/callbacks/test_finetuning_scheduler_callback.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -412,10 +412,10 @@ def test_finetuningscheduling_misconfiguration(tmpdir, callbacks: List[Callback]
412412
@pytest.mark.parametrize(
413413
"strategy, gpus, plugins",
414414
[
415-
pytest.param("ddp2", 2, None, marks=RunIf(min_gpus=2)),
416-
pytest.param("ddp_fully_sharded", 2, None, marks=RunIf(min_gpus=2)),
417-
pytest.param("horovod", None, None, marks=RunIf(min_gpus=2)),
418-
pytest.param("ddp", 2, "deepspeed_stage_2", marks=RunIf(deepspeed=True, min_gpus=2)),
415+
pytest.param("ddp2", 1, None, marks=RunIf(min_gpus=1)),
416+
pytest.param("ddp_fully_sharded", 1, None, marks=RunIf(min_gpus=1)),
417+
pytest.param("horovod", None, None, marks=RunIf(min_gpus=1)),
418+
pytest.param("deepspeed_stage_2", 1, None, marks=RunIf(deepspeed=True, min_gpus=1)),
419419
],
420420
)
421421
def test_finetuningscheduling_distributed_compat(tmpdir, strategy, gpus, plugins):

0 commit comments

Comments
 (0)