如何使tf.app.run脚本成为可导入模块

Question

tensorflow脚本通常具有以下结构。 如何将脚本转换成可导入的模块，以便可以从另一个python脚本调用主函数？

flags = tf.app.flags
flags.DEFINE_string('master', '', 'Name of the TensorFlow master to use.')
flags.DEFINE_integer('task', 0, 'task id')
flags.DEFINE_integer('num_clones', 1, 'Number of clones to deploy per worker.')
....

def main(_):
   ...

if __name__ == '__main__':
  tf.app.run()