From e0cd65c2b69a04b5cd7c348c6b80fdec1eabecf0 Mon Sep 17 00:00:00 2001
From: Xiaoyu Zhang <35585791+BBuf@users.noreply.github.com>
Date: Fri, 24 Jan 2025 00:33:59 +0800
Subject: [PATCH] [hotfix] fix test_sampling_scaling_penalties.py ci test
 (#3084)

---
 sgl-kernel/tests/test_sampling_scaling_penalties.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/sgl-kernel/tests/test_sampling_scaling_penalties.py b/sgl-kernel/tests/test_sampling_scaling_penalties.py
index 00f12bfbe..6194c7617 100644
--- 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__])