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

for more information, see https://pre-commit.ci
deepmodeling · Dec 24, 2024 · f7cf9e1 · f7cf9e1
1 parent d097bfa
commit f7cf9e1
Showing 1 changed file with 4 additions and 12 deletions.
diff --git a/deepmd/pt/model/model/property_model.py b/deepmd/pt/model/model/property_model.py
@@ -62,12 +62,8 @@ def forward(
             do_atomic_virial=do_atomic_virial,
         )
         model_predict = {}
-        model_predict[f"atom_{self.get_var_name()}"] = model_ret[
-            self.get_var_name()
-        ]
-        model_predict[self.get_var_name()] = model_ret[
-            f"{self.get_var_name()}_redu"
-        ]
+        model_predict[f"atom_{self.get_var_name()}"] = model_ret[self.get_var_name()]
+        model_predict[self.get_var_name()] = model_ret[f"{self.get_var_name()}_redu"]
         if "mask" in model_ret:
             model_predict["mask"] = model_ret["mask"]
         return model_predict
@@ -111,12 +107,8 @@ def forward_lower(
             extra_nlist_sort=self.need_sorted_nlist_for_lower(),
         )
         model_predict = {}
-        model_predict[f"atom_{self.get_var_name()}"] = model_ret[
-            self.get_var_name()
-        ]
-        model_predict[self.get_var_name()] = model_ret[
-            f"{self.get_var_name()}_redu"
-        ]
+        model_predict[f"atom_{self.get_var_name()}"] = model_ret[self.get_var_name()]
+        model_predict[self.get_var_name()] = model_ret[f"{self.get_var_name()}_redu"]
         if "mask" in model_ret:
             model_predict["mask"] = model_ret["mask"]
         return model_predict