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
...@@ -32,5 +32,4 @@ def test_sampling_scaling_penalties(batch_size, vocab_size, dtype): ...@@ -32,5 +32,4 @@ def test_sampling_scaling_penalties(batch_size, vocab_size, dtype):
if __name__ == "__main__": if __name__ == "__main__":
test_sampling_scaling_penalties() pytest.main([__file__])
print("All tests passed!")
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