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

for more information, see https://pre-commit.ci
deepmodeling · Jan 10, 2025 · 8763165 · 8763165
1 parent c2dc7ef
commit 8763165
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/deepmd/pt/entrypoints/main.py b/deepmd/pt/entrypoints/main.py
@@ -474,8 +474,8 @@ def change_bias(
             data_single.systems,
             data_single.dataloaders,
             nbatches,
-            min_frames_per_element_forstat = min_frames,
-            enable_element_completion = not skip_elem_check,
+            min_frames_per_element_forstat=min_frames,
+            enable_element_completion=not skip_elem_check,
         )
         updated_model = training.model_change_out_bias(
             model_to_change, sampled_data, _bias_adjust_mode=bias_adjust_mode

diff --git a/deepmd/utils/argcheck.py b/deepmd/utils/argcheck.py
@@ -2787,9 +2787,7 @@ def training_args(
         "If the gradient norm exceeds this value, it will be clipped to this limit. "
         "No gradient clipping will occur if set to 0."
     )
-    doc_min_frames_per_element_forstat = (
-        "The minimum number of frames per element used for statistics when using the mixed type." 
-    )
+    doc_min_frames_per_element_forstat = "The minimum number of frames per element used for statistics when using the mixed type."
     doc_enable_element_completion = (
         "Whether to check elements when using the mixed type"
     )