feat: Add support for Hub in model attach functionality

Author: Malav Shastri

src/sagemaker/jumpstart/model.py

@@ -429,6 +429,7 @@ def attach(
         cls,
         endpoint_name: str,
         inference_component_name: Optional[str] = None,
+        hub_name: Optional[str] = None,
         model_id: Optional[str] = None,
         model_version: Optional[str] = None,
         sagemaker_session=DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
@@ -457,6 +458,7 @@ def attach(
             model_id=model_id,
             model_version=model_version,
             sagemaker_session=sagemaker_session,
+            hub_name=hub_name,
         )
         model.endpoint_name = endpoint_name
         model.inference_component_name = inference_component_name

tests/unit/sagemaker/jumpstart/model/test_model.py

@@ -1360,6 +1360,7 @@ def test_attach(
             model_id="model-id",
             model_version="model-version",
             sagemaker_session=DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
+            hub_name=None,
         )
         assert isinstance(val, JumpStartModel)