SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Make transformers-cli cross-platform (#4131)

Browse Source 
* make transformers-cli cross-platform

Using "scripts" is a useful option in setup.py particularly when you want to get access to non-python scripts. However, in this case we want to have an entry point into some of our own Python scripts. To do this in a concise, cross-platfom way, we can use entry_points.console_scripts. This change is necessary to provide the CLI on different platforms, which "scripts" does not ensure. Usage remains the same, but the "transformers-cli" script has to be moved (be part of the library) and renamed (underscore + extension)

* make style & quality
This commit is contained in:
Bram Vanroy
2020-05-26 16:00:51 +02:00
committed by GitHub
parent c589eae2b8
commit 8cc6807e89
2 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@@ -127,7 +127,9 @@ setup(
"sacremoses", "sacremoses",
], ],
extras_require=extras, extras_require=extras,
scripts=["transformers-cli"], entry_points={
"console_scripts": ["transformers-cli=transformers.commands.transformers_cli:main"]
},
python_requires=">=3.6.0", python_requires=">=3.6.0",
classifiers=[ classifiers=[
"Development Status :: 5 - Production/Stable", "Development Status :: 5 - Production/Stable",

13
transformers-cli → src/transformers/commands/transformers_cli.py Executable file → Normal file
View File

@@ -8,9 +8,10 @@ from transformers.commands.run import RunCommand
from transformers.commands.serving import ServeCommand from transformers.commands.serving import ServeCommand
from transformers.commands.user import UserCommands from transformers.commands.user import UserCommands
if __name__ == '__main__':
parser = ArgumentParser('Transformers CLI tool', usage='transformers-cli <command> [<args>]') def main():
commands_parser = parser.add_subparsers(help='transformers-cli command helpers') parser = ArgumentParser("Transformers CLI tool", usage="transformers-cli <command> [<args>]")
commands_parser = parser.add_subparsers(help="transformers-cli command helpers")
# Register commands # Register commands
ConvertCommand.register_subcommand(commands_parser) ConvertCommand.register_subcommand(commands_parser)
@@ -23,10 +24,14 @@ if __name__ == '__main__':
# Let's go # Let's go
args = parser.parse_args() args = parser.parse_args()
if not hasattr(args, 'func'): if not hasattr(args, "func"):
parser.print_help() parser.print_help()
exit(1) exit(1)
# Run # Run
service = args.func(args) service = args.func(args)
service.run() service.run()
if __name__ == "__main__":
main()