[hotfix] fix test_sampling_scaling_penalties.py ci test (#3084)

2025-01-24 00:33:59 +08:00
parent f1b6861828
commit e0cd65c2b6
1 changed files with 1 additions and 2 deletions
--- a/sgl-kernel/tests/test_sampling_scaling_penalties.py
+++ b/sgl-kernel/tests/test_sampling_scaling_penalties.py
@@ -32,5 +32,4 @@ def test_sampling_scaling_penalties(batch_size, vocab_size, dtype):


 if __name__ == "__main__":
-    test_sampling_scaling_penalties()
-    print("All tests passed!")
+    pytest.main([__file__])