Skip to content

Commit aa2a08a

Browse files
committed
Update test
1 parent 1cd2da8 commit aa2a08a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/accelerators/test_accelerator_connector.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -447,10 +447,10 @@ def on_fit_start(self, trainer, pl_module):
447447
@mock.patch("pytorch_lightning.utilities._IS_INTERACTIVE", return_value=True)
448448
@mock.patch("torch.cuda.device_count", return_value=2)
449449
def test_ipython_incompatible_backend_error(*_):
450-
with pytest.raises(MisconfigurationException, match=r"strategy='ddp'\)` is not compatible"):
450+
with pytest.raises(MisconfigurationException, match=r"strategy='ddp'\)`.*is not compatible"):
451451
Trainer(accelerator="ddp", gpus=2)
452452

453-
with pytest.raises(MisconfigurationException, match=r"strategy='ddp2'\)` is not compatible"):
453+
with pytest.raises(MisconfigurationException, match=r"strategy='ddp2'\)`.*is not compatible"):
454454
Trainer(accelerator="ddp2", gpus=2)
455455

456456

0 commit comments

Comments
 (0)