From 0604d16fd02813d37efcc3ff270ae16b23af2ad3 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 1 Apr 2024 07:15:50 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 deepmd/pt/train/training.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/deepmd/pt/train/training.py b/deepmd/pt/train/training.py
index 7fcdea4efb..0f2046af25 100644
--- a/deepmd/pt/train/training.py
+++ b/deepmd/pt/train/training.py
@@ -1052,7 +1052,9 @@ def get_data(self, is_train=True, task_key="Default"):
                 except StopIteration:
                     # Refresh the status of the dataloader to start from a new epoch
                     with torch.device("cpu"):
-                        self.training_data[task_key] = self.training_dataloader[task_key]
+                        self.training_data[task_key] = self.training_dataloader[
+                            task_key
+                        ]
                     batch_data = next(iter(self.training_data[task_key]))
             else:
                 if self.validation_data[task_key] is None: