Fix pytest tests to exit with proper exit code (#16681)

2026-01-07 15:20:46 -08:00
parent 6037267f5b
commit 3a4767daa3
8 changed files with 22 additions and 8 deletions
@@ -144,4 +144,6 @@ def mixer2_gated_norm_tensor_parallel(


 if __name__ == "__main__":
-    pytest.main([__file__])
+    import sys
+
+    sys.exit(pytest.main([__file__]))