diff --git a/test/conftest.py b/test/conftest.py index f48cefc..67400aa 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -20,10 +20,12 @@ """ from __future__ import annotations +import numpy as np import pytest import torch # avoid randomness and non-deterministic algorithms +np.random.seed(0) torch.manual_seed(0) torch.use_deterministic_algorithms(True)