fix codestyle

Author: Malav Shastri

src/sagemaker/jumpstart/accessors.py

@@ -300,7 +300,9 @@ def get_model_specs(
                 return model_specs
 
             except Exception as ex:
-                logging.info("Recieved expection while calling APIs for ContentType Model: " + str(ex))
+                logging.info(
+                    "Recieved expection while calling APIs for ContentType Model: " + str(ex)
+                )
                 hub_model_arn = construct_hub_model_arn_from_inputs(
                     hub_arn=hub_arn, model_name=model_id, version=version
                 )

src/sagemaker/jumpstart/factory/model.py

@@ -34,7 +34,10 @@
     JUMPSTART_LOGGER,
 )
 from sagemaker.model_card.model_card import ModelCard, ModelPackageModelCard
-from sagemaker.jumpstart.hub.utils import construct_hub_model_arn_from_inputs, construct_hub_model_reference_arn_from_inputs
+from sagemaker.jumpstart.hub.utils import (
+    construct_hub_model_arn_from_inputs,
+    construct_hub_model_reference_arn_from_inputs,
+)
 from sagemaker.model_metrics import ModelMetrics
 from sagemaker.metadata_properties import MetadataProperties
 from sagemaker.drift_check_baselines import DriftCheckBaselines
@@ -552,15 +555,11 @@ def _add_tags_to_kwargs(kwargs: JumpStartModelDeployKwargs) -> Dict[str, Any]:
     if kwargs.hub_arn:
         if kwargs.model_reference_arn:
             hub_content_arn = construct_hub_model_reference_arn_from_inputs(
-                kwargs.hub_arn,
-                kwargs.model_id,
-                kwargs.model_version
+                kwargs.hub_arn, kwargs.model_id, kwargs.model_version
             )
         else:
             hub_content_arn = construct_hub_model_arn_from_inputs(
-                kwargs.hub_arn,
-                kwargs.model_id,
-                kwargs.model_version
+                kwargs.hub_arn, kwargs.model_id, kwargs.model_version
             )
         kwargs.tags = add_hub_content_arn_tags(kwargs.tags, hub_content_arn=hub_content_arn)
 

src/sagemaker/jumpstart/hub/hub.py

@@ -82,29 +82,25 @@ def __init__(
         self._sagemaker_session = sagemaker_session
         self.hub_storage_location = self._generate_hub_storage_location(bucket_name)
 
-    def _fetch_hub_bucket_name(self) -> str:
+    def _fetch_hub_bucket_name(self) -> Optional[str]:
         """Retrieves hub bucket name from Hub config if exists"""
         try:
             hub_response = self._sagemaker_session.describe_hub(hub_name=self.hub_name)
             hub_output_location = hub_response["S3StorageConfig"].get("S3OutputPath")
             if hub_output_location:
                 location = create_s3_object_reference_from_uri(hub_output_location)
                 return location.bucket
-            default_bucket_name = generate_default_hub_bucket_name(self._sagemaker_session)
             JUMPSTART_LOGGER.warning(
-                "There is not a Hub bucket associated with %s. Using %s",
+                "There is not a Hub bucket associated with %s.",
                 self.hub_name,
-                default_bucket_name,
             )
-            return default_bucket_name
+            return None
         except exceptions.ClientError:
-            hub_bucket_name = generate_default_hub_bucket_name(self._sagemaker_session)
             JUMPSTART_LOGGER.warning(
-                "There is not a Hub bucket associated with %s. Using %s",
+                "There is not a Hub bucket associated with %s.",
                 self.hub_name,
-                hub_bucket_name,
             )
-            return hub_bucket_name
+            return None
 
     def _generate_hub_storage_location(self, bucket_name: Optional[str] = None) -> None:
         """Generates an ``S3ObjectLocation`` given a Hub name."""