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

for more information, see https://pre-commit.ci
dingye18 · Dec 17, 2023 · 3545e76 · 3545e76
1 parent fe739eb
commit 3545e76
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/deepmd/entrypoints/train.py b/deepmd/entrypoints/train.py
@@ -505,15 +505,13 @@ def update_one_sel(jdata, descriptor, one_type: bool = False):
 
 
 def update_sel(jdata):
-    if jdata['model']['descriptor']['type'] != "se_a_mask":
-        log.info(
-            "Skip neighbor statistics for se_a_mask descriptor."
-        )
+    if jdata["model"]["descriptor"]["type"] != "se_a_mask":
+        log.info("Skip neighbor statistics for se_a_mask descriptor.")
         jdata_cpy = jdata.copy()
     else:
         log.info(
             "Calculate neighbor statistics... (add --skip-neighbor-stat to skip this step)"
         )
         jdata_cpy = jdata.copy()
-        jdata_cpy["model"] = Model.update_sel(jdata, jdata["model"])        
+        jdata_cpy["model"] = Model.update_sel(jdata, jdata["model"])
     return jdata_cpy