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
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -160,7 +160,7 @@ def main(args):
                    ))

        # update running id counter
        start_id += len(results)
        start_id += len(inputs)


def cli_main():