(self)
| 291 | return self |
| 292 | |
| 293 | def _put_indices(self): |
| 294 | assert self.batches_outstanding < 2 * self.num_workers |
| 295 | indices = next(self.sample_iter, None) |
| 296 | if indices is None: |
| 297 | return |
| 298 | self.index_queue.put((self.send_idx, indices)) |
| 299 | self.batches_outstanding += 1 |
| 300 | self.send_idx += 1 |
| 301 | |
| 302 | def _process_next_batch(self, batch): |
| 303 | self.rcvd_idx += 1 |
no test coverage detected