orbit.StandardEvaluator minor update

PiperOrigin-RevId: 329355546

orbit.StandardEvaluator minor update
PiperOrigin-RevId: 329355546
09050ac3 · Le Hou · A. Unique TensorFlower · 32309b8b · 09050ac3
Commit 09050ac3 authored Aug 31, 2020 by Le Hou Committed by A. Unique TensorFlower Aug 31, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

orbit/standard_runner.py orbit/standard_runner.py +1 -1

No files found.
--- a/orbit/standard_runner.py
+++ b/orbit/standard_runner.py
@@ -193,7 +193,7 @@ class StandardEvaluator(runner.AbstractEvaluator, metaclass=abc.ABCMeta):
    """See base class."""
    outputs = self.eval_begin()  # pylint: disable=assignment-from-no-return

-    eval_iter = tf.nest.map_structure(iter, self._eval_dataset)
+    eval_iter = tf.nest.map_structure(iter, self.eval_dataset)
    if self._eval_loop_fn is None:
      eval_fn = self.eval_step
      if self._eval_use_tf_function: