test: update linting

Author: bencrabtree

doc/conf.py

@@ -17,10 +17,11 @@
 from datetime import datetime
 import sys
 import os
-sys.path.append(os.path.join(os.path.dirname(__file__), '/doc/'))
 from jumpstart.utils import create_jumpstart_model_table
 
-project = u"sagemaker"
+sys.path.append(os.path.join(os.path.dirname(__file__), "/doc/"))
+
+project = "sagemaker"
 version = pkg_resources.require(project)[0].version
 
 # Add any Sphinx extension module names here, as strings. They can be extensions
@@ -42,7 +43,7 @@
 source_suffix = ".rst"  # The suffix of source filenames.
 master_doc = "index"  # The master toctree document.
 
-copyright = u"%s, Amazon" % datetime.now().year
+copyright = "%s, Amazon" % datetime.now().year
 
 # The full version, including alpha/beta/rc tags.
 release = version
@@ -80,7 +81,7 @@
 ]
 
 html_css_files = [
-    'https://cdn.datatables.net/1.10.23/css/jquery.dataTables.min.css',
+    "https://cdn.datatables.net/1.10.23/css/jquery.dataTables.min.css",
 ]
 
 html_context = {"css_files": ["_static/theme_overrides.css"]}
@@ -94,5 +95,6 @@
 # autosectionlabel
 autosectionlabel_prefix_document = True
 
+
 def setup(app):
-    create_jumpstart_model_table()
+    create_jumpstart_model_table()

doc/jumpstart/utils.py

@@ -1,3 +1,4 @@
+from __future__ import absolute_import
 import boto3
 import json
 from packaging.version import Version
@@ -21,6 +22,7 @@ def get_jumpstart_sdk_spec(key):
     spec = s3_client.get_object(Bucket=JUMPSTART_BUCKET, Key=key)["Body"].read().decode("utf-8")
     return json.loads(spec)
 
+
 def create_jumpstart_model_table():
     sdk_manifest = get_jumpstart_sdk_manifest()
     sdk_manifest_top_versions_for_models = {}
@@ -29,36 +31,47 @@ def create_jumpstart_model_table():
         if model["model_id"] not in sdk_manifest_top_versions_for_models:
             sdk_manifest_top_versions_for_models[model["model_id"]] = model
         else:
-            if Version(sdk_manifest_top_versions_for_models[model["model_id"]]["version"]) < Version(
-                model["version"]
-            ):
+            if Version(
+                sdk_manifest_top_versions_for_models[model["model_id"]]["version"]
+            ) < Version(model["version"]):
                 sdk_manifest_top_versions_for_models[model["model_id"]] = model
 
     f = open("jumpstart/jumpstart.rst", "w")
 
-    f.write('==================================\n')
-    f.write('JumpStart Available Model Table\n')
-    f.write('==================================\n')
-    f.write('\n')
-    f.write('JumpStart for the SageMaker Python SDK uses model ids and model versions to access the necessary utilities. This table serves to provide the core material plus some extra information that can be useful in selecting the correct model id and corresponding parameters.\n')
-    f.write('\n')
-    f.write('If you want to automatically use the latest version of the model, use "*" for the `model_version` attribute. We highly suggest pinning an exact model version however.\n')
-    f.write('\n')
-    f.write('.. list-table:: Available Models\n')
-    f.write('   :widths: 50 20 20 20\n')
-    f.write('   :header-rows: 1\n')
-    f.write('   :class: datatable\n')
-    f.write('\n')
-    f.write('   * - Model ID\n')
-    f.write('     - Fine Tunable?\n')
-    f.write('     - Latest Version\n')
-    f.write('     - Min SDK Version\n')
+    f.write("==================================\n")
+    f.write("JumpStart Available Model Table\n")
+    f.write("==================================\n")
+    f.write("\n")
+    f.write(
+        """
+    JumpStart for the SageMaker Python SDK uses model ids and model versions to access the necessary
+    utilities. This table serves to provide the core material plus some extra information that can be useful
+    in selecting the correct model id and corresponding parameters.\n
+    """
+    )
+    f.write("\n")
+    f.write(
+        """
+    If you want to automatically use the latest version of the model, use "*" for the `model_version` attribute.
+    We highly suggest pinning an exact model version however.\n
+    """
+    )
+    f.write("\n")
+    f.write(".. list-table:: Available Models\n")
+    f.write("   :widths: 50 20 20 20\n")
+    f.write("   :header-rows: 1\n")
+    f.write("   :class: datatable\n")
+    f.write("\n")
+    f.write("   * - Model ID\n")
+    f.write("     - Fine Tunable?\n")
+    f.write("     - Latest Version\n")
+    f.write("     - Min SDK Version\n")
 
     for model in sorted(sdk_manifest, key=lambda elt: elt["model_id"]):
         model_spec = get_jumpstart_sdk_spec(model["spec_key"])
-        f.write('   * - {}\n'.format(model["model_id"]))
-        f.write('     - {}\n'.format(model_spec["training_supported"]))
-        f.write('     - {}\n'.format(model["version"]))
-        f.write('     - {}\n'.format(model["min_version"]))
+        f.write("   * - {}\n".format(model["model_id"]))
+        f.write("     - {}\n".format(model_spec["training_supported"]))
+        f.write("     - {}\n".format(model["version"]))
+        f.write("     - {}\n".format(model["min_version"]))
 
-    f.close()
+    f.close()