Skip to content

Commit a20ddee

Browse files
author
Han Wang
committed
fix bug of nopbc in dp test
1 parent 637692b commit a20ddee

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

source/train/test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,8 @@ def test_ener (args) :
5252

5353
coord = test_data["coord"][:numb_test].reshape([numb_test, -1])
5454
box = test_data["box"][:numb_test]
55+
if not data.pbc:
56+
box = None
5557
atype = test_data["type"][0]
5658
if dp.get_dim_fparam() > 0:
5759
fparam = test_data["fparam"][:numb_test]

0 commit comments

Comments
 (0)