From d5303281989597367d517c462498309551900690 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 5 Dec 2023 10:33:34 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- pvnet/data/wind_datamodule.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pvnet/data/wind_datamodule.py b/pvnet/data/wind_datamodule.py index 7c1b4a45..c0f3476a 100644 --- 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 = (