Skip to content

GitLab

Unverified Commit e0cd65c2 authored by Xiaoyu Zhang's avatar Xiaoyu Zhang Committed by GitHub
Browse files

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

parent f1b68618
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
sgl-kernel/tests/test_sampling_scaling_penalties.py
View file @ e0cd65c2
......@@ -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__])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment