Commit 829bd8ce authored by Myle Ott's avatar Myle Ott Committed by Facebook Github Bot
Browse files

Add standalone binaries 

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

Differential Revision: D13956810

Pulled By: myleott

fbshipit-source-id: 61ace179d1d3790226c38b3f3e47f5452b5ec514
parent bbb4120b

eval_lm.py

+5 −1
Original line number Diff line number Diff line
@@ -174,7 +174,11 @@ def main(parsed_args): 
            print(ws)





if __name__ == '__main__':

def cli_main():

    parser = options.get_eval_lm_parser()

    args = options.parse_args_and_arch(parser)

    main(args)





if __name__ == '__main__':

    cli_main()

generate.py

+5 −1
Original line number Diff line number Diff line
@@ -178,7 +178,11 @@ def main(args): 
        print('| Generate {} with beam={}: {}'.format(args.gen_subset, args.beam, scorer.result_string()))





if __name__ == '__main__':

def cli_main():

    parser = options.get_generation_parser()

    args = options.parse_args_and_arch(parser)

    main(args)





if __name__ == '__main__':

    cli_main()

interactive.py

+5 −1
Original line number Diff line number Diff line
@@ -183,7 +183,11 @@ def main(args): 
                    print(align)





if __name__ == '__main__':

def cli_main():

    parser = options.get_generation_parser(interactive=True)

    args = options.parse_args_and_arch(parser)

    main(args)





if __name__ == '__main__':

    cli_main()

preprocess.py

+5 −1
Original line number Diff line number Diff line
@@ -271,7 +271,11 @@ def merge_files(files, outpath): 
    ds.finalize("{}.idx".format(outpath))





if __name__ == "__main__":

def cli_main():

    parser = options.get_preprocessing_parser()

    args = parser.parse_args()

    main(args)





if __name__ == "__main__":

    cli_main()

setup.py

+10 −0
Original line number Diff line number Diff line
@@ -43,4 +43,14 @@ setup( 
    packages=find_packages(),

    ext_modules=[bleu],

    test_suite='tests',

    entry_points={

        'console_scripts': [

            'fairseq-eval-lm = eval_lm:cli_main',

            'fairseq-generate = generate:cli_main',

            'fairseq-interactive = interactive:cli_main',

            'fairseq-preprocess = preprocess:cli_main',

            'fairseq-train = train:cli_main',

            'fairseq-score = score:main',

        ],

    },

)