Skip to content

Commit 2748a84

Browse files
committed
Bring back num_batches
1 parent 5ae83bc commit 2748a84

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

bayesflow/datasets/online_dataset.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,3 +90,7 @@ def __getitem__(self, item: int) -> dict[str, np.ndarray]:
9090
batch = self.adapter(batch, stage=self.stage)
9191

9292
return batch
93+
94+
@property
95+
def num_batches(self) -> int:
96+
return self._num_batches

0 commit comments

Comments
 (0)