remove useless comments

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
deepmodeling · Oct 26, 2024 · 2e3f3ad · 2e3f3ad
1 parent d35198b
commit 2e3f3ad
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/deepmd/jax/infer/deep_eval.py b/deepmd/jax/infer/deep_eval.py
@@ -382,8 +382,6 @@ def _get_output_shape(self, odef, nframes, natoms):
             return [nframes, *odef.shape[:-1], natoms, 3]
         elif odef.category == OutputVariableCategory.OUT:
             # atom_energy, atom_tensor
-            # Something wrong here?
-            # return [nframes, *shape, natoms, 1]
             return [nframes, natoms, *odef.shape, 1]
         else:
             raise RuntimeError("unknown category")