Yield next batch of data
(self, input_data: Callable)
| 340 | self.it = 0 |
| 341 | |
| 342 | def next(self, input_data: Callable) -> bool: |
| 343 | """Yield next batch of data""" |
| 344 | if self.it == len(self._data): |
| 345 | # Return False when there's no more batch. |
| 346 | return False |
| 347 | input_data(data=self.data(), label=self.labels()) |
| 348 | self.it += 1 |
| 349 | return True |
| 350 | |
| 351 | |
| 352 | @pytest.mark.skipif(**tm.no_cudf()) |