def execute(self, context): hook = MLEngineHook( gcp_conn_id=self._gcp_conn_id, delegate_to=self._delegate_to) if self._operation == 'create': return hook.create_model(project_id=self._project_id, model=self._model) elif self._operation == 'get': return hook.get_model(project_id=self._project_id, model_name=self._model['name']) else: raise ValueError('Unknown operation: {}'.format(self._operation))
def execute(self, context): hook = MLEngineHook(gcp_conn_id=self._gcp_conn_id, delegate_to=self._delegate_to) return hook.create_model(project_id=self._project_id, model=self._model)