def after_run(self, run_context: tf.estimator.SessionRunContext,
run_values: tf.estimator.SessionRunValues) -> None:
if run_context.stop_requested:
run_context._stop_requested = False
self.context.set_stop_requested(True)
def after_run(self, run_context: tf.estimator.SessionRunContext,
run_values: tf.estimator.SessionRunValues) -> None:
run_context.request_stop()