def after_run(self, run_context, run_values): # Adapt run_context and run_values to OnRunEndRequest and invoke superclass # on_run_end() if self._performed_action == framework.OnRunStartAction.DEBUG_RUN: on_run_end_request = framework.OnRunEndRequest( self._performed_action, run_values.run_metadata) self.on_run_end(on_run_end_request)
def after_run(self, run_context, run_values): # Adapt run_context and run_values to OnRunEndRequest and invoke superclass # on_run_end() on_run_end_request = framework.OnRunEndRequest(self._performed_action, run_values.run_metadata) self._session_wrapper.on_run_end(on_run_end_request)