Commit 2340832f authored by Myle Ott's avatar Myle Ott Committed by Facebook Github Bot
Browse files

Output original IDs in interactive.py (in case some rows are filtered; fixes #591)

Summary: Pull Request resolved: https://github.com/pytorch/fairseq/pull/606

Differential Revision: D14680968

Pulled By: myleott

fbshipit-source-id: 8044d828a8167199c10f2aee24f7e611feb91802
parent a78ad1ac
...@@ -160,7 +160,7 @@ def main(args): ...@@ -160,7 +160,7 @@ def main(args):
)) ))
# update running id counter # update running id counter
start_id += len(results) start_id += len(inputs)
def cli_main(): def cli_main():
......
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