From 4dd84fc32baab4a8486f083f546afadc28d37014 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Thu, 25 Jan 2024 11:00:31 -0500 Subject: [PATCH] fix seed Signed-off-by: Jinzhe Zeng --- tests/test_force_grad.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_force_grad.py b/tests/test_force_grad.py index 72ee2d5..191f62a 100644 --- a/tests/test_force_grad.py +++ b/tests/test_force_grad.py @@ -95,7 +95,7 @@ def get_dataset(self, system_index=0, batch_index=0): def test_force_grad(self, threshold=1e-3, delta0=1e-6, seed=20): result0 = self.model(**get_data(self.origin_batch)) - np.random.default_rng().seed(seed) + np.random.default_rng(seed) errors = np.zeros((self.dpdatasystem._natoms, 3)) for atom_index in range(self.dpdatasystem._natoms): for axis_index in range(3):