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

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

deepmd/dpmodel/utils/serialization.py

@@ -106,9 +106,9 @@ def save_dp_model(filename: str, model_dict: dict) -> None:
         with h5py.File(filename, "w") as f:
             model_dict = traverse_model_dict(
                 model_dict,
-                lambda x: f.create_dataset(
-                    f"variable_{variable_counter():04d}", data=x
-                ).name,
+                lambda x: (
+                    f.create_dataset(f"variable_{variable_counter():04d}", data=x).name
+                ),
             )
             save_dict = {
                 **extra_dict,
@@ -118,15 +118,17 @@ def save_dp_model(filename: str, model_dict: dict) -> None:
     elif filename_extension in {".yaml", ".yml"}:
         model_dict = traverse_model_dict(
             model_dict,
-            lambda x: {
-                "@class": "np.ndarray",
-                "@is_variable": True,
-                "@version": 1,
-                "dtype": x.dtype.name,
-                "value": x.tolist(),
-            }
-            if isinstance(x, np.ndarray)
-            else x,
+            lambda x: (
+                {
+                    "@class": "np.ndarray",
+                    "@is_variable": True,
+                    "@version": 1,
+                    "dtype": x.dtype.name,
+                    "value": x.tolist(),
+                }
+                if isinstance(x, np.ndarray)
+                else x
+            ),
         )
         with open(filename, "w") as f:
             yaml.safe_dump(

deepmd/pd/utils/multi_task.py

@@ -150,8 +150,10 @@ def replace_one_item(
     for shared_key in shared_links:
         shared_links[shared_key]["links"] = sorted(
             shared_links[shared_key]["links"],
-            key=lambda x: x["shared_level"]
-            - ("spin" in model_config["model_dict"][x["model_key"]]) * 100,
+            key=lambda x: (
+                x["shared_level"]
+                - ("spin" in model_config["model_dict"][x["model_key"]]) * 100
+            ),
         )
         # little trick to make spin models in the front to be the base models,
         # because its type embeddings are more general.

deepmd/pt/utils/multi_task.py

@@ -155,8 +155,10 @@ def replace_one_item(
     for shared_key in shared_links:
         shared_links[shared_key]["links"] = sorted(
             shared_links[shared_key]["links"],
-            key=lambda x: x["shared_level"]
-            - ("spin" in model_config["model_dict"][x["model_key"]]) * 100,
+            key=lambda x: (
+                x["shared_level"]
+                - ("spin" in model_config["model_dict"][x["model_key"]]) * 100
+            ),
         )
         # little trick to make spin models in the front to be the base models,
         # because its type embeddings are more general.