linter

Author: bencrabtree

src/sagemaker/jumpstart/curated_hub/curated_hub.py

@@ -14,6 +14,7 @@
 from __future__ import absolute_import
 
 from typing import Optional, Dict, Any
+from sagemaker.jumpstart.constants import DEFAULT_JUMPSTART_SAGEMAKER_SESSION
 
 from sagemaker.session import Session
 

src/sagemaker/jumpstart/estimator.py

@@ -529,7 +529,9 @@ def _is_valid_model_id_hook():
             if match:
                 hub_arn = hub_name
             else:
-                hub_arn = construct_hub_arn_from_name(hub_name=hub_name, region=region, sagemaker_session=sagemaker_session)
+                hub_arn = construct_hub_arn_from_name(
+                    hub_name=hub_name, region=region, sagemaker_session=sagemaker_session
+                )
 
         estimator_init_kwargs = get_init_kwargs(
             model_id=model_id,

src/sagemaker/jumpstart/types.py

@@ -1748,7 +1748,6 @@ def __init__(
         self.skip_model_validation = skip_model_validation
 
 
-
 class HubArnExtractedInfo(JumpStartDataHolderType):
     """Data class for info extracted from Hub arn."""
 

src/sagemaker/jumpstart/utils.py

@@ -878,7 +878,7 @@ def construct_hub_arn_from_name(
     session: Optional[Session] = None,
 ) -> str:
     """Constructs a Hub arn from the Hub name using default Session values"""
-    print('being called')
+    print("being called")
 
     if not session:
         session = constants.DEFAULT_JUMPSTART_SAGEMAKER_SESSION
@@ -894,6 +894,7 @@ def construct_hub_model_arn_from_inputs(hub_arn: str, model_name: str, version:
     """Constructs a HubContent model arn from the Hub name, model name, and model version"""
 
     info = extract_info_from_hub_content_arn(hub_arn)
-    arn = f"arn:{info.partition}:sagemaker:{info.region}:{info.account_id}:hub-content/{info.hub_name}/Model/{model_name}/{version}"
+    arn = f"arn:{info.partition}:sagemaker:{info.region}:{info.account_id}:hub-content/" \
+        f"{info.hub_name}/Model/{model_name}/{version}"
 
     return arn

tests/unit/sagemaker/jumpstart/estimator/test_estimator.py

@@ -34,7 +34,6 @@
 from sagemaker.jumpstart.enums import JumpStartScriptScope, JumpStartTag
 
 from sagemaker.jumpstart.estimator import JumpStartEstimator
-from sagemaker.jumpstart.types import JumpStartEstimatorFitKwargs
 
 from sagemaker.jumpstart.utils import get_jumpstart_content_bucket
 from sagemaker.session_settings import SessionSettings

tests/unit/sagemaker/jumpstart/utils.py

@@ -153,7 +153,7 @@ def get_spec_from_base_spec(
 
     if version and semantic_version_str:
         raise ValueError("Cannot specify both `version` and `semantic_version_str` fields.")
-    
+
     if model_id is not None:
         if all(
             [