diff --git a/tests/pytest_random_plugin.py b/tests/pytest_random_plugin.py index 8726e2c..0f5071f 100644 --- a/tests/pytest_random_plugin.py +++ b/tests/pytest_random_plugin.py @@ -1,4 +1,5 @@ import hashlib +import random import pytest import torch @@ -15,4 +16,5 @@ def seed(request): ).hexdigest(), 16) % (2**31) seed = base + node_hash torch.manual_seed(seed) + random.seed(seed) return seed