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)
