8c6df6412ae7141f46e7752be2a2a16c519b8ccc,official/nlp/tasks/question_answering.py,QuestionAnsweringTask,aggregate_logs,#QuestionAnsweringTask#Any#Any#,241

Before Change

            self.raw_aggregated_result(
                unique_id=values[0],
                start_logits=values[1].tolist(),
                end_logits=values[2].tolist()))
    return state

  def reduce_aggregated_logs(self, aggregated_logs):

After Change

    for outputs in zip(step_outputs["unique_ids"],
                       step_outputs["start_logits"],
                       step_outputs["end_logits"]):
      numpy_values = [
          output.numpy() for output in outputs if output is not None]

      for values in zip(*numpy_values):
        state.append(RawAggregatedResult(
            unique_id=values[0],
            start_logits=values[1],

Instances