fix no attribute '_FlagValues' issue

supersational · Jun 23, 2018 · 5cce00f · 5cce00f
1 parent 9db1c98
commit 5cce00f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/seq2seq/test/pipeline_test.py b/seq2seq/test/pipeline_test.py
@@ -41,7 +41,9 @@
 def _clear_flags():
   """Resets Tensorflow's FLAG values"""
   #pylint: disable=W0212
-  tf.app.flags.FLAGS = tf.app.flags._FlagValues()
+  for flag_key in dir(tf.app.flags.FLAGS):
+      delattr(tf.app.flags.FLAGS, flag_key)
+  #tf.app.flags.FLAGS = tf.app.flags._FlagValues()
   tf.app.flags._global_parser = argparse.ArgumentParser()