CsvPipelineDataFormat: Fix for single-column

55397dfb · Julien Chaumond · 2fde5a24 · 55397dfb
Commit 55397dfb authored Dec 17, 2019 by Julien Chaumond
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

transformers/pipelines.py transformers/pipelines.py +1 -1

No files found.
--- a/transformers/pipelines.py
+++ b/transformers/pipelines.py
@@ -136,7 +136,7 @@ class CsvPipelineDataFormat(PipelineDataFormat):
                if self.is_multi_columns:
                    yield {k: row[c] for k, c in self.column}
                else:
-                    yield row[self.column]
+                    yield row[self.column[0]]

    def save(self, data: List[dict]):
        with open(self.output, 'w') as f: