[hotfix] fix test_sampling_scaling_penalties.py ci test (#3084)
This commit is contained in:
@@ -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!")
|
|
||||||
|
|||||||
Reference in New Issue
Block a user