riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Fix leading axis added when saving through the command run

Browse Source
This commit is contained in:
Morgan Funtowicz 2019-12-20 14:47:04 +01:00
parent c37815f130
commit 7f74084528
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
transformers/commands/run.py
View File

@ -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: