orbit.StandardEvaluator minor update

PiperOrigin-RevId: 329355546
ashishk98 · Aug 31, 2020 · 09050ac · 09050ac
1 parent 32309b8
commit 09050ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/orbit/standard_runner.py b/orbit/standard_runner.py
@@ -193,7 +193,7 @@ def evaluate(
     """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: