From 8f38dbdddca3b0a5d0739e8194fd3afeb510f1ba Mon Sep 17 00:00:00 2001 From: Benjamin Gallusser <bgallusser@googlemail.com> Date: Mon, 17 Jun 2024 13:50:22 +0200 Subject: [PATCH] Bring back mps tests --- tests/test_pretrained.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_pretrained.py b/tests/test_pretrained.py index 6515805..b948d1b 100644 --- a/tests/test_pretrained.py +++ b/tests/test_pretrained.py @@ -9,7 +9,7 @@ @pytest.mark.parametrize("name", ["ctc", "general_2d"]) -@pytest.mark.parametrize("device", ["cpu", "cuda"]) +@pytest.mark.parametrize("device", ["cpu", "cuda", "mps"]) def test_pretrained(name, device): """Each pretrained model should run on all (available) device.""" # TODO mps support