address pylint

Author: laurenyu

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

@@ -22,7 +22,7 @@
 
 
 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,7 +51,7 @@ 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 --in-file foo.py --out-file bar.py"
@@ -72,5 +72,6 @@ def _parse_args():
 
 
 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