def before_run(self, run_context):
    del run_context
    return session_run_hook.SessionRunArgs(fetches=None, feed_dict=self.feed_fn)
def before_run(self, run_context):
    del run_context
    return session_run_hook.SessionRunArgs(fetches=None, feed_dict=self.feed_fn)