Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Aug 28, 2024
1 parent fabf9e1 commit aff1eef
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion deepmd/utils/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -629,7 +629,9 @@ def _load_data(
densities.append(batch_densities)
grids = np.concatenate(grids) # [ngrids, 3]
densities = np.concatenate(densities) # [ngrids]
grid_densities = np.concatenate([grids, densities.reshape([-1, 1])], axis=1) # [ngrids, 4]
grid_densities = np.concatenate(
[grids, densities.reshape([-1, 1])], axis=1
) # [ngrids, 4]
data.append(grid_densities)
data = np.stack(data) # [nframes, ngrids, 4]
return np.float32(1.0), data
Expand Down

0 comments on commit aff1eef

Please sign in to comment.