Removed __main__
This commit is contained in:
5
setup.py
5
setup.py
@@ -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',
|
||||||
|
|||||||
@@ -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 <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()
|
|
||||||
Reference in New Issue
Block a user