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

Fix leading axis added when saving through the command run

parent c37815f1
......@@ -50,9 +50,9 @@ class RunCommand(BaseTransformersCLICommand):
nlp, output = self._nlp, []
for entry in self._reader:
if self._reader.is_multi_columns:
output += [nlp(**entry)]
output += nlp(**entry)
else:
output += [nlp(entry)]
output += nlp(entry)
# Saving data
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