add hub_arn support for accept_types, content_types, serializers, deserializers, and predictor (aws#4463)

Author: bencrabtree

src/sagemaker/jumpstart/factory/model.py

@@ -725,6 +725,7 @@ def get_init_kwargs(
     model_version: Optional[str] = None,
     hub_arn: Optional[str] = None,
     model_type: Optional[JumpStartModelType] = JumpStartModelType.OPEN_WEIGHTS,
+    hub_arn: Optional[str] = None,
     tolerate_vulnerable_model: Optional[bool] = None,
     tolerate_deprecated_model: Optional[bool] = None,
     instance_type: Optional[str] = None,
@@ -758,6 +759,7 @@ def get_init_kwargs(
         model_version=model_version,
         hub_arn=hub_arn,
         model_type=model_type,
+        hub_arn=hub_arn,
         instance_type=instance_type,
         region=region,
         image_uri=image_uri,

src/sagemaker/jumpstart/types.py

@@ -1256,6 +1256,7 @@ class JumpStartModelInitKwargs(JumpStartKwargs):
         "model_version",
         "hub_arn",
         "model_type",
+        "hub_arn",
         "instance_type",
         "tolerate_vulnerable_model",
         "tolerate_deprecated_model",
@@ -1288,6 +1289,7 @@ class JumpStartModelInitKwargs(JumpStartKwargs):
         "model_version",
         "hub_arn",
         "model_type",
+        "hub_arn",
         "tolerate_vulnerable_model",
         "tolerate_deprecated_model",
         "region",
@@ -1301,6 +1303,7 @@ def __init__(
         model_version: Optional[str] = None,
         hub_arn: Optional[str] = None,
         model_type: Optional[JumpStartModelType] = JumpStartModelType.OPEN_WEIGHTS,
+        hub_arn: Optional[str] = None,
         region: Optional[str] = None,
         instance_type: Optional[str] = None,
         image_uri: Optional[Union[str, Any]] = None,
@@ -1332,6 +1335,7 @@ def __init__(
         self.model_version = model_version
         self.hub_arn = hub_arn
         self.model_type = model_type
+        self.hub_arn = hub_arn
         self.instance_type = instance_type
         self.region = region
         self.image_uri = image_uri