Override the data service configs of params from flags.
(self, params)
| 346 | return params |
| 347 | |
| 348 | def parse_data_service(self, params): |
| 349 | """Override the data service configs of params from flags.""" |
| 350 | if ('tf_data_service' in self._flags_obj and |
| 351 | self._flags_obj.tf_data_service and |
| 352 | isinstance(params.task, config_definitions.TaskConfig)): |
| 353 | params.override({ |
| 354 | 'task': { |
| 355 | 'train_data': { |
| 356 | 'tf_data_service_address': self._flags_obj.tf_data_service, |
| 357 | }, |
| 358 | 'validation_data': { |
| 359 | 'tf_data_service_address': self._flags_obj.tf_data_service, |
| 360 | } |
| 361 | } |
| 362 | }) |
| 363 | return params |
| 364 | |
| 365 | def parse_params_override(self, params): |
| 366 | # Get the second level of override from `--params_override`. |