Skip to content

GitLab

Commit 7f740845 authored by Morgan Funtowicz's avatar Morgan Funtowicz
Browse files

Fix leading axis added when saving through the command run

parent c37815f1
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
transformers/commands/run.py
View file @ 7f740845
...@@ -50,9 +50,9 @@ class RunCommand(BaseTransformersCLICommand): ...@@ -50,9 +50,9 @@ class RunCommand(BaseTransformersCLICommand):
nlp, output = self._nlp, [] nlp, output = self._nlp, []
for entry in self._reader: for entry in self._reader:
if self._reader.is_multi_columns: if self._reader.is_multi_columns:
output += [nlp(**entry)] output += nlp(**entry)
else: else:
output += [nlp(entry)] output += nlp(entry)
# Saving data # Saving data
if self._nlp.binary_output: if self._nlp.binary_output:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment