(self)
| 444 | self._batch_feeder = batch_feeder |
| 445 | |
| 446 | def run(self): |
| 447 | input_data = self._worker_fun(self._worker_id, self._batch_size) |
| 448 | |
| 449 | self._batch_feeder.put(input_data, self._coordinator) |
| 450 | |
| 451 | def finish(self): |
| 452 | self._metrics.put_metric( |