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

for more information, see https://pre-commit.ci
deepmodeling · Apr 1, 2024 · 0604d16 · 0604d16
1 parent 8b1b280
commit 0604d16
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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: