def _push_next(self):
r = next(self._iter, None)<EOL>if r is None:<EOL><INDENT>return<EOL><DEDENT>async_ret = self._worker_pool.apply_async(<EOL>self._worker_fn, (r, self._batchify_fn, self._dataset))<EOL>self._data_buffer[self._sent_idx] = async_ret<EOL>self._sent_idx += <NUM_LIT:1><EOL>
Assign next batch workload to workers.