def testDoWithGenericExecutorClass(self):
executor = ai_platform_trainer_executor.GenericExecutor()
executor.Do(self._inputs, self._outputs, self._exec_properties)
self.mock_runner.start_aip_training.assert_called_with(
self._inputs, self._outputs, self._exec_properties,
self._generic_executor_class_path, {
'project': self._project_id,
'jobDir': self._job_dir,
}, None)
def testDoWithGenericExecutorClass(self):
executor = ai_platform_trainer_executor.GenericExecutor()
executor.Do(self._inputs, self._outputs,
self._serialize_custom_config_under_test())
self.mock_runner.start_cloud_training.assert_called_with(
self._inputs, self._outputs,
self._serialize_custom_config_under_test(),
self._generic_executor_class_path, {
'project': self._project_id,
'jobDir': self._job_dir,
}, None, False, None)