Skip to content

Commit bff8f16

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent bf8bd13 commit bff8f16

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

dpdata/abacus/stru.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -731,7 +731,9 @@ def process_file_input(file_input, atom_names, input_name):
731731
out += "0.0\n"
732732
out += str(data["atom_numbs"][iele]) + "\n"
733733
for iatom in range(data["atom_numbs"][iele]):
734-
iatomtype = np.nonzero(data["atom_types"] == iele)[0][iatom] # it is the atom index
734+
iatomtype = np.nonzero(data["atom_types"] == iele)[0][
735+
iatom
736+
] # it is the atom index
735737
iout = f"{data['coords'][frame_idx][iatomtype, 0]:.12f} {data['coords'][frame_idx][iatomtype, 1]:.12f} {data['coords'][frame_idx][iatomtype, 2]:.12f}"
736738
# add flags for move, velocity, mag, angle1, angle2, and sc
737739
if move is not None:

0 commit comments

Comments
 (0)