Skip to content

Commit 21d9089

Browse files
committed
fix test
1 parent b180eec commit 21d9089

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
@@ -929,9 +929,9 @@ def test_unsupported_ipu_choice(mock_ipu_acc_avail, monkeypatch):
929929

930930
monkeypatch.setattr(imports, "_IPU_AVAILABLE", True)
931931
monkeypatch.setattr(ipu, "_IPU_AVAILABLE", True)
932-
with pytest.raises(MisconfigurationException, match=r"accelerator='ipu', precision='bf16'\)` is not supported"):
932+
with pytest.raises(ValueError, match=r"accelerator='ipu', precision='bf16'\)` is not supported"):
933933
Trainer(accelerator="ipu", precision="bf16")
934-
with pytest.raises(MisconfigurationException, match=r"accelerator='ipu', precision=64\)` is not supported"):
934+
with pytest.raises(ValueError, match=r"accelerator='ipu', precision=64\)` is not supported"):
935935
Trainer(accelerator="ipu", precision=64)
936936

937937

0 commit comments

Comments
 (0)