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