Improved documentation.

PiperOrigin-RevId: 203776243
STCYC · Jul 9, 2018 · deee57e · deee57e
1 parent fdcb7ca
commit deee57e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/python/estimator/training.py b/tensorflow/python/estimator/training.py
@@ -312,10 +312,10 @@ def train_and_evaluate(estimator, train_spec, eval_spec):
   #       hidden_units=[1024, 512, 256])
 
   # Input pipeline for train and evaluate.
-  def train_input_fn: # returns x, y
+  def train_input_fn(): # returns x, y
     # please shuffle the data.
     pass
-  def eval_input_fn_eval: # returns x, y
+  def eval_input_fn(): # returns x, y
     pass
 
   train_spec = tf.estimator.TrainSpec(input_fn=train_input_fn, max_steps=1000)