eb09eb97d28c918756c12a8960d5daa85d098abc,app/server/api.py,CSVHandler,parse,#CSVHandler#Any#,354

Before Change

        reader = csv.reader(file)
        columns = None
        for i, row in enumerate(reader, start=1):
            if i == 1:                           // skip header
                columns = row
                continue
            elif len(row) == len(columns) == 2:  // text with a label
                text, label = row
                yield text, label
            else:
                raise FileParseException(line_num=i, line=row)

    def render(self):
        raise NotImplementedError()

After Change

        for i, row in enumerate(reader, start=2):
            if len(row) == len(columns) and len(row) >= 2:
                text, label = row[:2]
                meta = json.dumps(dict(zip(columns[2:], row[2:])))
                data = {"text": text, "meta": meta}
                yield data, label
            else:
                raise FileParseException(line_num=i, line=row)

Instances