From 54133271c9d7b89a41015695cff5074853ccb24c Mon Sep 17 00:00:00 2001 From: vfdev Date: Mon, 5 Aug 2024 21:46:18 +0200 Subject: [PATCH] Fix mps ci --- tests/ignite/engine/test_create_supervised.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/ignite/engine/test_create_supervised.py b/tests/ignite/engine/test_create_supervised.py index 6f7e0a2b218..54938167601 100644 --- a/tests/ignite/engine/test_create_supervised.py +++ b/tests/ignite/engine/test_create_supervised.py @@ -316,7 +316,8 @@ def _test_create_supervised_evaluator( # This is broken in 1.6.0 but will be probably fixed with 1.7.0 err_msg_1 = "Expected all tensors to be on the same device" err_msg_2 = "Placeholder storage has not been allocated on MPS device" - with pytest.raises(RuntimeError, match=f"({err_msg_1}|{err_msg_2})"): + err_msg_3 = "Tensor for argument weight is on cpu but expected on mps" + with pytest.raises(RuntimeError, match=f"({err_msg_1}|{err_msg_2}|{err_msg_3})"): evaluator.run(data)