"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "72d363d9791c64eff21ed7535c690de2c636d508"
Commit a305067f authored by Morgan Funtowicz's avatar Morgan Funtowicz
Browse files

Removed __main__

parent 3492a6ec
...@@ -68,11 +68,6 @@ setup( ...@@ -68,11 +68,6 @@ setup(
scripts=[ scripts=[
'transformers-cli' 'transformers-cli'
], ],
entry_points={
'console_scripts': [
"transformers=transformers.__main__:main",
]
},
# python_requires='>=3.5.0', # python_requires='>=3.5.0',
classifiers=[ classifiers=[
'Intended Audience :: Science/Research', 'Intended Audience :: Science/Research',
......
# 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 <command> [<args>]')
# 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()
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