diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py
index bdbee19108..b8d13e6f25 100644
--- 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