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

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