def step_function(model, iterator):
      """Runs a single evaluation step."""

      def run_step(data):
        outputs = model.test_step(data)