diff --git a/credit/datasets/era5_multistep_batcher.py b/credit/datasets/era5_multistep_batcher.py index f13ca28..bbf7a3c 100644 --- a/credit/datasets/era5_multistep_batcher.py +++ b/credit/datasets/era5_multistep_batcher.py @@ -337,7 +337,7 @@ def set_epoch(self, epoch): self.initialize_batch() def batches_per_epoch(self): - return math.ceil((self.batch_indices) / self.batch_size) + return math.ceil(len(self.batch_indices) / self.batch_size) def __getitem__(self, _): """