REVERT: hardcode torch.mps

Author: ev-br

array_api_tests/__init__.py

@@ -9,6 +9,7 @@
 
 __all__ = ["xp", "api_version", "xps"]
 
+"""
 
 # You can comment the following out and instead import the specific array module
 # you want to test, e.g. `import array_api_strict as xp`.
@@ -40,6 +41,12 @@
         "ARRAY_API_TESTS_MODULE environment variable."
     )
 
+"""
+
+import array_api_compat.torch as xp
+xp_name = xp.__name__
+
+xp.set_default_device("mps")
 
 # If xp.bool is not available, like in some versions of NumPy and CuPy, try
 # patching in xp.bool_.