20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Bugfix] fix flaky test (#13089)

Browse Source 
Signed-off-by: மனோஜ்குமார் பழனிச்சாமி <smartmanoj42857@gmail.com>
This commit is contained in:
மனோஜ்குமார் பழனிச்சாமி 2025-02-11 20:11:20 +05:30 committed by GitHub
parent 2e3b969ec0
commit 110f59a33e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/test_seed_behavior.py
View File

@ -8,32 +8,17 @@ from vllm.platforms.interface import Platform
def test_seed_behavior(): def test_seed_behavior():
# Test with seed=None # Test with a specific seed
Platform.seed_everything(None) Platform.seed_everything(42)
random_value_1 = random.randint(0, 100) random_value_1 = random.randint(0, 100)
np_random_value_1 = np.random.randint(0, 100) np_random_value_1 = np.random.randint(0, 100)
torch_random_value_1 = torch.randint(0, 100, (1, )).item() torch_random_value_1 = torch.randint(0, 100, (1, )).item()
Platform.seed_everything(None) Platform.seed_everything(42)
random_value_2 = random.randint(0, 100) random_value_2 = random.randint(0, 100)
np_random_value_2 = np.random.randint(0, 100) np_random_value_2 = np.random.randint(0, 100)
torch_random_value_2 = torch.randint(0, 100, (1, )).item() torch_random_value_2 = torch.randint(0, 100, (1, )).item()
assert random_value_1 != random_value_2 assert random_value_1 == random_value_2
assert np_random_value_1 != np_random_value_2 assert np_random_value_1 == np_random_value_2
assert torch_random_value_1 != torch_random_value_2 assert torch_random_value_1 == torch_random_value_2
# Test with a specific seed
Platform.seed_everything(42)
random_value_3 = random.randint(0, 100)
np_random_value_3 = np.random.randint(0, 100)
torch_random_value_3 = torch.randint(0, 100, (1, )).item()
Platform.seed_everything(42)
random_value_4 = random.randint(0, 100)
np_random_value_4 = np.random.randint(0, 100)
torch_random_value_4 = torch.randint(0, 100, (1, )).item()
assert random_value_3 == random_value_4
assert np_random_value_3 == np_random_value_4
assert torch_random_value_3 == torch_random_value_4