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

for more information, see https://pre-commit.ci
deepmodeling · Feb 29, 2024 · d5a1549 · d5a1549
1 parent 66edca5
commit d5a1549
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py
@@ -167,7 +167,9 @@ def get_dataloader_and_buffer(_data, _params):
                     validation_dataloader,
                     validation_data_buffered,
                 ) = get_dataloader_and_buffer(_validation_data, _training_params)
-                valid_numb_batch = _training_params["validation_data"].get("numb_btch", 1)
+                valid_numb_batch = _training_params["validation_data"].get(
+                    "numb_btch", 1
+                )
             else:
                 validation_dataloader = None
                 validation_data_buffered = None