Skip to content

Commit

Permalink
Merge branch 'master' into update-pth-versions
Browse files Browse the repository at this point in the history
  • Loading branch information
vfdev-5 authored Aug 5, 2024
2 parents b680b0c + e58039e commit 6ce3f45
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/ignite/engine/test_create_supervised.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)


Expand Down

0 comments on commit 6ce3f45

Please sign in to comment.