add python testing for cuda with ability to skip list of tests (#2295)

2025-06-15 10:56:48 -07:00
parent 580776559b
commit 4fda5fbdf9
36 changed files with 220 additions and 35 deletions
@@ -2033,4 +2033,4 @@ class TestArray(mlx_tests.MLXTestCase):


 if __name__ == "__main__":
-    unittest.main()
+    mlx_tests.MLXTestRunner()