diff --git a/eval_lm.py b/eval_lm.py index be7d967f1..0422cb6c7 100644 --- a/eval_lm.py +++ b/eval_lm.py @@ -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() diff --git a/generate.py b/generate.py index cc258a48f..21902cb9f 100644 --- a/generate.py +++ b/generate.py @@ -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() diff --git a/interactive.py b/interactive.py index ad60bd532..6e93bd33a 100644 --- a/interactive.py +++ b/interactive.py @@ -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() diff --git a/preprocess.py b/preprocess.py index e66b577d2..19554c1da 100644 --- a/preprocess.py +++ b/preprocess.py @@ -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() diff --git a/setup.py b/setup.py index 53e8eaf51..c0483252b 100644 --- a/setup.py +++ b/setup.py @@ -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', + ], + }, ) diff --git a/train.py b/train.py index 6110191e1..413619461 100644 --- a/train.py +++ b/train.py @@ -380,7 +380,7 @@ def distributed_main(i, args): main(args, init_distributed=True) -if __name__ == '__main__': +def cli_main(): parser = options.get_training_parser() args = options.parse_args_and_arch(parser) @@ -415,3 +415,7 @@ if __name__ == '__main__': else: # single GPU training main(args) + + +if __name__ == '__main__': + cli_main()