seed

Signed-off-by: Jinzhe Zeng <[email protected]>
dptech-corp · Jan 25, 2024 · 1cdca9e · 1cdca9e
1 parent 60d6566
commit 1cdca9e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_force_grad.py b/tests/test_force_grad.py
@@ -95,6 +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)
         errors = np.zeros((self.dpdatasystem._natoms, 3))
         for atom_index in range(self.dpdatasystem._natoms):
             for axis_index in range(3):