Skip to content

Commit c4e112a

Browse files
authored
Merge pull request #196 from njzjz/patch-2
support nopbc
2 parents ec3fcff + 71b42c9 commit c4e112a

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

dpgen/generator/run.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -430,17 +430,21 @@ def run_train (iter_index,
430430
for single_sys in os.listdir(os.path.join(ii)):
431431
trans_comm_data += glob.glob(os.path.join(ii, single_sys, 'set.*'))
432432
trans_comm_data += glob.glob(os.path.join(ii, single_sys, 'type.raw'))
433+
trans_comm_data += glob.glob(os.path.join(ii, single_sys, 'nopbc'))
433434
else:
434435
trans_comm_data += glob.glob(os.path.join(ii, 'set.*'))
435436
trans_comm_data += glob.glob(os.path.join(ii, 'type.raw'))
437+
trans_comm_data += glob.glob(os.path.join(ii, 'nopbc'))
436438
for ii in fp_data :
437439
if jdata.get('use_clusters', False):
438440
for single_sys in os.listdir(os.path.join(ii)):
439441
trans_comm_data += glob.glob(os.path.join(ii, single_sys, 'set.*'))
440442
trans_comm_data += glob.glob(os.path.join(ii, single_sys, 'type.raw'))
443+
trans_comm_data += glob.glob(os.path.join(ii, single_sys, 'nopbc'))
441444
else:
442445
trans_comm_data += glob.glob(os.path.join(ii, 'set.*'))
443446
trans_comm_data += glob.glob(os.path.join(ii, 'type.raw'))
447+
trans_comm_data += glob.glob(os.path.join(ii, 'nopbc'))
444448
os.chdir(cwd)
445449

446450
try:

0 commit comments

Comments
 (0)