Skip to content

Commit

Permalink
fix seed
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
  • Loading branch information
njzjz committed Jan 25, 2024
1 parent 1cdca9e commit 4dd84fc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_force_grad.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 4dd84fc

Please sign in to comment.