riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 18:22:34 +06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge pull request #130 from sodre/use-entry-points

Browse Source 
Use entry-points instead of scripts
This commit is contained in:
Thomas Wolf 2018-12-19 10:18:24 +01:00 committed by GitHub
commit 2feb29c0ff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bin/pytorch_pretrained_bert
View File

@ -1,2 +0,0 @@
#!/bin/sh
python -m pytorch_pretrained_bert "$@"

5
pytorch_pretrained_bert/__main__.py
View File

@ -1,5 +1,5 @@
# coding: utf8 # coding: utf8
if __name__ == '__main__': def main():
import sys import sys
try: try:
from .convert_tf_checkpoint_to_pytorch import convert_tf_checkpoint_to_pytorch from .convert_tf_checkpoint_to_pytorch import convert_tf_checkpoint_to_pytorch
@ -17,3 +17,6 @@ if __name__ == '__main__':
TF_CONFIG = sys.argv.pop() TF_CONFIG = sys.argv.pop()
TF_CHECKPOINT = sys.argv.pop() TF_CHECKPOINT = sys.argv.pop()
convert_tf_checkpoint_to_pytorch(TF_CHECKPOINT, TF_CONFIG, PYTORCH_DUMP_OUTPUT) convert_tf_checkpoint_to_pytorch(TF_CHECKPOINT, TF_CONFIG, PYTORCH_DUMP_OUTPUT)
if __name__ == '__main__':
main()

6
setup.py
View File

@ -53,7 +53,11 @@ setup(
'boto3', 'boto3',
'requests', 'requests',
'tqdm'], 'tqdm'],
scripts=["bin/pytorch_pretrained_bert"], entry_points={
'console_scripts': [
"pytorch_pretrained_bert=pytorch_pretrained_bert.__main__:main"
]
},
python_requires='>=3.5.0', python_requires='>=3.5.0',
tests_require=['pytest'], tests_require=['pytest'],
classifiers=[ classifiers=[