Commit edcd29f2 authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Replace the two remaining instances of parser.parse_args() for better compatibility (#2590)

parent 487d18e2
...@@ -59,5 +59,5 @@ def main(unused_argv): ...@@ -59,5 +59,5 @@ def main(unused_argv):
if __name__ == '__main__': if __name__ == '__main__':
FLAGS = parser.parse_args() FLAGS, unparsed = parser.parse_known_args()
tf.app.run() tf.app.run(argv=[sys.argv[0]] + unparsed)
...@@ -20,6 +20,7 @@ from __future__ import print_function ...@@ -20,6 +20,7 @@ from __future__ import print_function
import argparse import argparse
import os import os
import sys
from six.moves import urllib from six.moves import urllib
import tensorflow as tf import tensorflow as tf
...@@ -31,6 +32,7 @@ EVAL_FILE = 'adult.test' ...@@ -31,6 +32,7 @@ EVAL_FILE = 'adult.test'
EVAL_URL = '%s/%s' % (DATA_URL, EVAL_FILE) EVAL_URL = '%s/%s' % (DATA_URL, EVAL_FILE)
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( parser.add_argument(
'--data_dir', type=str, default='/tmp/census_data', '--data_dir', type=str, default='/tmp/census_data',
help='Directory to download census data') help='Directory to download census data')
...@@ -65,5 +67,5 @@ def main(unused_argv): ...@@ -65,5 +67,5 @@ def main(unused_argv):
if __name__ == '__main__': if __name__ == '__main__':
FLAGS = parser.parse_args() FLAGS, unparsed = parser.parse_known_args()
tf.app.run() tf.app.run(argv=[sys.argv[0]] + unparsed)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment