Make sure tests are ran with pytest (#1538)

pytorch · Jan 13, 2025 · de5c6e1 · de5c6e1
1 parent 12396c6
commit de5c6e1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/test_ops.py b/test/test_ops.py
@@ -6,7 +6,6 @@
     TestCase,
     instantiate_parametrized_tests,
     parametrize,
-    run_tests,
 )
 from torch.testing._internal.optests import opcheck
 
@@ -615,4 +614,4 @@ def test_marlin_qqq(batch_size, k_chunk, n_chunk, num_bits, group_size, mnk_fact
 
 
 if __name__ == "__main__":
-    run_tests()
+    pytest.main([__file__])