From a305067f2d6ca74865c6d686608a1428e476a32f Mon Sep 17 00:00:00 2001 From: Morgan Funtowicz Date: Thu, 19 Dec 2019 19:41:48 +0100 Subject: [PATCH] Removed __main__ --- setup.py | 5 ----- transformers/__main__.py | 36 ------------------------------------ 2 files changed, 41 deletions(-) delete mode 100644 transformers/__main__.py diff --git a/setup.py b/setup.py index b3b6e2e063..56c4d1733b 100644 --- a/setup.py +++ b/setup.py @@ -68,11 +68,6 @@ setup( scripts=[ 'transformers-cli' ], - entry_points={ - 'console_scripts': [ - "transformers=transformers.__main__:main", - ] - }, # python_requires='>=3.5.0', classifiers=[ 'Intended Audience :: Science/Research', diff --git a/transformers/__main__.py b/transformers/__main__.py deleted file mode 100644 index a6e9ae65e0..0000000000 --- a/transformers/__main__.py +++ /dev/null @@ -1,36 +0,0 @@ -# coding: utf8 - -def main(): - import sys - if len(sys.argv) < 2 or sys.argv[1] not in ["convert", "train", "predict", "serve"]: - print( - "First argument to `transformers` command line interface should be one of: \n" - ">> convert serve train predict") - if sys.argv[1] == "convert": - from transformers.commands import convert - convert(sys.argv) - elif sys.argv[1] == "train": - from transformers.commands import train - train(sys.argv) - elif sys.argv[1] == "serve": - pass - # from argparse import ArgumentParser - # from transformers.commands.serving import ServeCommand - # parser = ArgumentParser('Transformers CLI tool', usage='transformers serve []') - # commands_parser = parser.add_subparsers(help='transformers-cli command helpers') - - # # Register commands - # ServeCommand.register_subcommand(commands_parser) - - # # Let's go - # args = parser.parse_args() - - # if not hasattr(args, 'func'): - # parser.print_help() - # exit(1) - # # Run - # service = args.func(args) - # service.run() - -if __name__ == '__main__': - main()