aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
authorGravatar Vijay Vasudevan <vrv@google.com>2016-02-23 11:29:49 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-02-23 11:31:50 -0800
commit91e1180fac8defe3cc5ea91509d3658d5a08f61a (patch)
tree3211e90f79c589aa1feebec7aff761cc83ee9f82
parent986762835b7484ded2893ebee967c77857d6b970 (diff)
TensorFlow gflags: add DEFINE_bool as an alias to DEFINE_boolean
Change: 115364038
-rw-r--r--tensorflow/python/platform/default/_flags.py5
-rw-r--r--tensorflow/python/platform/default/flags_test.py2
2 files changed, 6 insertions, 1 deletions
diff --git a/tensorflow/python/platform/default/_flags.py b/tensorflow/python/platform/default/_flags.py
index 66ff3a89dc..5ba011f9c1 100644
--- a/tensorflow/python/platform/default/_flags.py
+++ b/tensorflow/python/platform/default/_flags.py
@@ -106,6 +106,11 @@ def DEFINE_boolean(flag_name, default_value, docstring):
dest=flag_name)
+# The internal google library defines the following alias, so we match
+# the API for consistency.
+DEFINE_bool = DEFINE_boolean # pylint: disable=invalid-name
+
+
def DEFINE_float(flag_name, default_value, docstring):
"""Defines a flag of type 'float'.
diff --git a/tensorflow/python/platform/default/flags_test.py b/tensorflow/python/platform/default/flags_test.py
index e6cd57d5a9..c057f96993 100644
--- a/tensorflow/python/platform/default/flags_test.py
+++ b/tensorflow/python/platform/default/flags_test.py
@@ -34,7 +34,7 @@ flags.DEFINE_boolean("bool_negation", True, "HelpString")
flags.DEFINE_boolean("bool_a", False, "HelpString")
flags.DEFINE_boolean("bool_c", False, "HelpString")
flags.DEFINE_boolean("bool_d", True, "HelpString")
-flags.DEFINE_boolean("bool_e", True, "HelpString")
+flags.DEFINE_bool("bool_e", True, "HelpString")
FLAGS = flags.FLAGS