Skip to content

Commit ff77dc6

Browse files
pee8379amcadmusfelix5572shazj99Zhengju Sha
authored
Fix QE pwscf AssertionError (#685)
* fix typo in github actions release to conda * Get from_poscar_path only when from_poscar is true (#537) Change-Id: I17774bee345634e4e72bd783e8112eefaaf9f0d3 Co-authored-by: Zhengju Sha <jenny@bytedance.com> * Fix QE pwscf AssertionError details in #683 Co-authored-by: Han Wang <amcadmus@gmail.com> Co-authored-by: felix5572 <felix5572@github.com> Co-authored-by: shazj99 <shazj99@gmail.com> Co-authored-by: Zhengju Sha <jenny@bytedance.com> Co-authored-by: Han Wang <wang_han@iapcm.ac.cn>
1 parent 0b36b59 commit ff77dc6

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

dpgen/generator/run.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2097,8 +2097,12 @@ def make_fp_pwscf(iter_index,
20972097
for ii in fp_tasks:
20982098
os.chdir(ii)
20992099
sys_data = dpdata.System('POSCAR').data
2100-
sys_data['atom_masses'] = jdata['mass_map']
2101-
ret = make_pwscf_input(sys_data, fp_pp_files, fp_params, user_input = user_input)
2100+
sys_data['atom_masses'] = []
2101+
pps = []
2102+
for iii in sys_data['atom_names']:
2103+
sys_data['atom_masses'].append(jdata['mass_map'][jdata['type_map'].index(iii)])
2104+
pps.append(fp_pp_files[jdata['type_map'].index(iii)])
2105+
ret = make_pwscf_input(sys_data, pps, fp_params, user_input = user_input)
21022106
with open('input', 'w') as fp:
21032107
fp.write(ret)
21042108
os.chdir(cwd)

0 commit comments

Comments
 (0)