seed random number generator

HERA-Team · Feb 13, 2025 · c8d57b5 · c8d57b5
1 parent 55b5d29
commit c8d57b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hera_filters/tests/test_dspec.py b/hera_filters/tests/test_dspec.py
@@ -1513,7 +1513,7 @@ def test_sparse_linear_fit_2d_non_binary_wgts():
     flags = (time_flags | freq_flags)
 
     # Generate separable, non-binary weights
-    axis_1_weights = (~time_flags[:, 0]).astype(float) * np.random.randint(1, 10, size=(ntimes, ))
+    axis_1_weights = (~time_flags[:, 0]).astype(float) * rng.randint(1, 10, size=(ntimes, ))
     axis_2_weights = (~freq_flags[0]).astype(float)
     wgts = np.outer(axis_1_weights, axis_2_weights)