Skip to content

Commit 7cf97c2

Browse files
committed
test: add minimum gpu requirement in test_deepspeed_summary_with_rich_model_summary
1 parent f5badec commit 7cf97c2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tests_pytorch/utilities/test_deepspeed_model_summary.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def on_fit_start(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule") -
5757
trainer.fit(model)
5858

5959

60-
@RunIf(deepspeed=True, rich=True)
60+
@RunIf(min_cuda_gpus=1, deepspeed=True, rich=True)
6161
@mock.patch("rich.table.Table.add_row", autospec=True)
6262
def test_deepspeed_summary_with_rich_model_summary(mock_table_add_row, tmp_path):
6363
from lightning.pytorch.callbacks import RichModelSummary

0 commit comments

Comments
 (0)