(self)
| 681 | self._shutdown() |
| 682 | |
| 683 | def _next_non_threaded(self): |
| 684 | batch = next(self.dataloader_it) |
| 685 | batch = recursive_apply( |
| 686 | batch, restore_parent_storage_columns, self.dataloader.graph |
| 687 | ) |
| 688 | batch, feats, stream_event = _prefetch( |
| 689 | batch, self.dataloader, self.stream |
| 690 | ) |
| 691 | return batch, feats, stream_event |
| 692 | |
| 693 | def _next_threaded(self): |
| 694 | try: |
no test coverage detected