Skip to content

Commit f1859eb

Browse files
authored
Fix bug in abacus MD support (#214)
* add ABACUS MD interface. * Update system.py * Add files via upload * Update md.py * Update md.py
1 parent 5990c78 commit f1859eb

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

dpdata/abacus/md.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def get_frame (fname):
164164
atom_names, natoms, types, coords = get_coords(celldm, cell, geometry_inlines, inlines)
165165
# This coords is not to be used.
166166
dump_freq = get_coord_dump_freq(inlines = inlines)
167-
ndump = int(os.popen("ls --l %s | grep 'md_pos_' | wc -l" %path_out).readlines()[0])
167+
ndump = int(os.popen("ls -l %s | grep 'md_pos_' | wc -l" %path_out).readlines()[0])
168168
# number of dumped geometry files
169169
coords = get_coords_from_cif(ndump, dump_freq, atom_names, natoms, types, path_out, cell)
170170

@@ -185,6 +185,8 @@ def get_frame (fname):
185185
data['energies'] = energy
186186
data['forces'] = force
187187
data['virials'] = stress
188+
if type(data['virials']) != np.ndarray:
189+
del data['virials']
188190
data['orig'] = np.zeros(3)
189191

190192
return data

0 commit comments

Comments
 (0)