(self)
| 707 | return batch, feats, stream_event |
| 708 | |
| 709 | def __next__(self): |
| 710 | batch, feats, stream_event = ( |
| 711 | self._next_non_threaded() |
| 712 | if not self.use_thread |
| 713 | else self._next_threaded() |
| 714 | ) |
| 715 | batch = recursive_apply_pair(batch, feats, _assign_for) |
| 716 | if stream_event is not None: |
| 717 | stream_event.wait() |
| 718 | return batch |
| 719 | |
| 720 | |
| 721 | # Make them classes to work with pickling in mp.spawn |
nothing calls this directly
no test coverage detected