Updated batches_per_epoch method

NCAR · Dec 20, 2024 · d67b2a4 · d67b2a4
1 parent 3452571
commit d67b2a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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, _):
         """