change: add v2 migration script to console_scripts in setup.py

setup.py

@@ -105,6 +105,11 @@ def read_version():
     ],
     install_requires=required_packages,
     extras_require=extras,
-    entry_points={"console_scripts": ["sagemaker=sagemaker.cli.main:main"]},
+    entry_points={
+        "console_scripts": [
+            "sagemaker=sagemaker.cli.main:main",
+            "sagemaker-upgrade-v2=sagemaker.cli.compatibility.v2.sagemaker_upgrade_v2:main",
+        ]
+    },
     include_package_data=True,  # TODO-reinvent-2019 [knakad]: Remove after rule_configs is in PyPI
 )

tools/compatibility/v2/ast_transformer.py → src/sagemaker/cli/compatibility/v2/ast_transformer.py

@@ -15,7 +15,7 @@
 
 import ast
 
-from modifiers import framework_version
+from sagemaker.cli.compatibility.v2.modifiers import framework_version
 
 FUNCTION_CALL_MODIFIERS = [framework_version.FrameworkVersionEnforcer()]
 

tools/compatibility/v2/files.py → src/sagemaker/cli/compatibility/v2/files.py

@@ -20,7 +20,7 @@
 
 import pasta
 
-from ast_transformer import ASTTransformer
+from sagemaker.cli.compatibility.v2.ast_transformer import ASTTransformer
 
 LOGGER = logging.getLogger(__name__)
 

tools/compatibility/v2/modifiers/framework_version.py → src/sagemaker/cli/compatibility/v2/modifiers/framework_version.py

@@ -15,7 +15,7 @@
 
 import ast
 
-from modifiers.modifier import Modifier
+from sagemaker.cli.compatibility.v2.modifiers.modifier import Modifier
 
 FRAMEWORK_DEFAULTS = {
     "Chainer": "4.1.0",

tools/compatibility/v2/sagemaker_upgrade_v2.py → src/sagemaker/cli/compatibility/v2/sagemaker_upgrade_v2.py

@@ -16,13 +16,13 @@
 import argparse
 import os
 
-import files
+from sagemaker.cli.compatibility.v2 import files
 
 _EXT_TO_UPDATER_CLS = {".py": files.PyFileUpdater, ".ipynb": files.JupyterNotebookFileUpdater}
 
 
 def _update_file(input_file, output_file):
-    """Update a file to be compatible with v2 of the SageMaker Python SDK,
+    """Updates a file to be compatible with v2 of the SageMaker Python SDK,
     and write the updated source to the output file.
 
     Args:
@@ -51,10 +51,10 @@ def _update_file(input_file, output_file):
 
 
 def _parse_args():
-    """Parses CLI arguments"""
+    """Parses CLI arguments."""
     parser = argparse.ArgumentParser(
         description="A tool to convert files to be compatible with v2 of the SageMaker Python SDK. "
-        "Simple usage: sagemaker_upgrade_v2.py --in-file foo.py --out-file bar.py"
+        "Simple usage: sagemaker-upgrade-v2 --in-file foo.py --out-file bar.py"
     )
     parser.add_argument(
         "--in-file",
@@ -71,6 +71,7 @@ def _parse_args():
     return parser.parse_args()
 
 
-if __name__ == "__main__":
+def main():
+    """Parses the CLI arguments and executes the file update."""
     args = _parse_args()
     _update_file(args.in_file, args.out_file)

tox.ini

@@ -82,7 +82,7 @@ skip_install = true
 deps =
     pylint==2.3.1
 commands =
-    python -m pylint --rcfile=.pylintrc -j 0 src/sagemaker tools
+    python -m pylint --rcfile=.pylintrc -j 0 src/sagemaker
 
 [testenv:twine]
 basepython = python3