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

for more information, see https://pre-commit.ci
openclimatefix · Dec 5, 2023 · d530328 · d530328
1 parent feb9f04
commit d530328
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pvnet/data/wind_datamodule.py b/pvnet/data/wind_datamodule.py
@@ -95,7 +95,9 @@ def _get_premade_batches_datapipe(self, subdir, shuffle=False):
             filenames=list(glob.glob(f"{self.batch_dir}/{subdir}/*.nc")),
         )
         data_pipeline = (
-            data_pipeline.batch(self.batch_size).map(stack_np_examples_into_batch).map(batch_to_tensor)
+            data_pipeline.batch(self.batch_size)
+            .map(stack_np_examples_into_batch)
+            .map(batch_to_tensor)
         )
         if shuffle:
             data_pipeline = (