We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2dac5ff commit 6946f7fCopy full SHA for 6946f7f
dpdata/deepmd/comp.py
@@ -33,6 +33,8 @@ def to_system_data(folder,
33
nframes = np.reshape(cells, [-1,3,3]).shape[0]
34
all_cells.append(np.reshape(cells, [nframes,3,3]))
35
all_coords.append(np.reshape(coords, [nframes,-1,3]))
36
+ if eners is not None:
37
+ eners = np.reshape(eners, [nframes])
38
if labels:
39
if eners is not None and len(eners) > 0:
40
all_eners.append(np.reshape(eners, [nframes]))
0 commit comments