(self)
| 226 | return (arg,) |
| 227 | |
| 228 | def reset_indices(self): |
| 229 | self.current_iter = 0 |
| 230 | self.current_sample = 0 |
| 231 | |
| 232 | def prefetch(self, pool, context_i, batch_size, epoch_idx): |
| 233 | # NOTE We can't schedule more than what's on top of pipeline's prefetch queue, as the |
no outgoing calls
no test coverage detected