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
This commit is contained in:
Patrick Sodré
2018-12-19 01:57:04 +00:00
parent d57763f582
commit 87c1244c7d
2 changed files with 9 additions and 2 deletions

View File

@@ -1,5 +1,5 @@
# coding: utf8
if __name__ == '__main__':
def main():
import sys
try:
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_CHECKPOINT = sys.argv.pop()
convert_tf_checkpoint_to_pytorch(TF_CHECKPOINT, TF_CONFIG, PYTORCH_DUMP_OUTPUT)
if __name__ == '__main__':
main()