Use a fixed seed for finite differences tests

lab-cosmo · Jul 3, 2024 · 8dd7adf · 8dd7adf
1 parent ef1383c
commit 8dd7adf
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/metatensor-operations/tests/add.py b/python/metatensor-operations/tests/add.py
@@ -276,7 +276,8 @@ def function(array):
  tensor_2 = _gradcheck.cartesian_cubic(array)
  return metatensor.add(tensor_1, tensor_2)
 
- array = np.random.rand(5, 3)
+ rng = np.random.default_rng(seed=123456)
+ array = rng.random((5, 3))
  _gradcheck.check_finite_differences(function, array, parameter="g")
 
 
@@ -287,5 +288,7 @@ def function(array):
  tensor_2 = _gradcheck.cartesian_cubic(array)
  return metatensor.add(tensor_1, tensor_2)
 
- array = torch.rand(5, 3, dtype=torch.float64)
+ rng = torch.Generator()
+ rng.manual_seed(123456)
+ array = torch.rand(5, 3, dtype=torch.float64, generator=rng)
  _gradcheck.check_finite_differences(function, array, parameter="g")