317da380d4f10c8bba55ee3dc5110079fad86ea7,pytext/metric_reporters/intent_slot_detection_metric_reporter.py,IntentSlotMetricReporter,batch_context,#IntentSlotMetricReporter#Any#Any#,207

Before Change

    def batch_context(self, raw_batch, batch):
        return {
            DatasetFieldName.UTTERANCE_FIELD: [row["text"] for row in raw_batch],
            DatasetFieldName.SEQ_LENS: batch["tokens"][1],
            DatasetFieldName.TOKEN_RANGE: batch["tokens"][2],
            DatasetFieldName.RAW_WORD_LABEL: [
                ",".join([str(x) for x in row[self.slot_column_name]])
                for row in raw_batch
            ],
            BatchContext.INDEX: [
                row[RawExampleFieldName.ROW_INDEX] for row in raw_batch
            ],

After Change

        context[DatasetFieldName.SEQ_LENS] = batch[self.token_tensorizer_name][
            1
        ].tolist()
        context[DatasetFieldName.TOKEN_RANGE] = batch[self.token_tensorizer_name][
            2
        ].tolist()
        context[DatasetFieldName.RAW_WORD_LABEL] = [
            self.get_raw_slot_str(raw_data_row) for raw_data_row in raw_batch
        ]
        return context

Instances