Commit 87c1244c authored by Patrick Sodré's avatar Patrick Sodré
Browse files

Convert scripts into entry_points

The recommended approach to create launch scripts is to use entry_points
and console_scripts.

xref: https://packaging.python.org/guides/distributing-packages-using-setuptools/#scripts
parent d57763f5
# 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__': ...@@ -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()
...@@ -53,7 +53,11 @@ setup( ...@@ -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=[
......
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