fix typo

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Feb 28, 2024 · 9f0b9b7 · 9f0b9b7
1 parent ced2a3a
commit 9f0b9b7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deepmd/tf/fit/ener.py b/deepmd/tf/fit/ener.py
@@ -304,8 +304,8 @@ def _compute_output_stats(self, all_stat, rcond=1e-3, mixed_type=False):
             sys_tynatom,
             assigned_bias=assigned_atom_ener,
             rcond=rcond,
-        ).ravel()
-        return energy_shift
+        )
+        return energy_shift.ravel()
 
     def compute_input_stats(self, all_stat: dict, protection: float = 1e-2) -> None:
         """Compute the input statistics.