[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
deepmodeling · Mar 6, 2024 · 17c04d2 · 17c04d2
1 parent 9d7fea7
commit 17c04d2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/deepmd/pt/loss/tensor.py b/deepmd/pt/loss/tensor.py
@@ -83,9 +83,7 @@ def forward(self, model_pred, label, natoms, learning_rate=0.0, mae=False):
             Other losses for display.
         """
         del learning_rate, mae
-        loss = torch.zeros(
-            1, dtype=env.GLOBAL_PT_FLOAT_PRECISION, device=env.DEVICE
-        )[0]
+        loss = torch.zeros(1, dtype=env.GLOBAL_PT_FLOAT_PRECISION, device=env.DEVICE)[0]
         more_loss = {}
         if (
             self.has_local_weight