(self)
| 414 | return len(self.datas) == 0 |
| 415 | |
| 416 | def __call__(self): |
| 417 | if not self.datas: |
| 418 | return [] |
| 419 | |
| 420 | if self.idx >= len(self.datas): |
| 421 | self.idx = 0 |
| 422 | if self.idx + self.real_batch_size > len(self.datas): |
| 423 | batch_data = self.datas[self.idx :] |
| 424 | self.idx = len(self.datas) |
| 425 | else: |
| 426 | batch_data = self.datas[self.idx : self.idx + self.real_batch_size] |
| 427 | self.idx += self.real_batch_size |
| 428 | |
| 429 | # Update absolute index for checkpoint |
| 430 | self.current_abs_idx = self.start_idx + self.idx |
| 431 | |
| 432 | # Load images for this batch on-demand |
| 433 | batch_data = self._load_images_for_batch(batch_data) |
| 434 | |
| 435 | return batch_data |
nothing calls this directly
no test coverage detected