fix integ test

Author: bencrabtree

src/sagemaker/jumpstart/hub/hub.py

@@ -94,20 +94,22 @@ def create(
         """Creates a hub with the given description"""
         curr_timestamp = datetime.now().timestamp()
 
-        return self._sagemaker_session.create_hub(
-            hub_name=self.hub_name,
-            hub_description=description,
-            hub_display_name=display_name,
-            hub_search_keywords=search_keywords,
-            s3_storage_config={
+        request = {
+            "hub_name": self.hub_name,
+            "hub_description": description,
+            "hub_display_name": display_name,
+            "hub_search_keywords": search_keywords,
+            "tags": tags
+        }
+        
+        if self.bucket_name:
+            request["s3_storage_config"] = {
                 "S3OutputPath": (
                     f"s3://{self.bucket_name}/{self.hub_name}-{curr_timestamp}"
-                    if self.bucket_name
-                    else None
                 )
-            },
-            tags=tags,
-        )
+            }
+
+        return self._sagemaker_session.create_hub(**request)
 
     def describe(self, hub_name: Optional[str] = None) -> DescribeHubResponse:
         """Returns descriptive information about the Hub"""

src/test.py

@@ -0,0 +1,24 @@
+import json
+import boto3
+from sagemaker.session import Session
+from sagemaker.jumpstart.model import JumpStartModel
+from sagemaker.jumpstart.estimator import JumpStartEstimator
+from sagemaker.enums import EndpointType
+from sagemaker.jumpstart.artifacts.predictors import _retrieve_default_content_type
+
+model_id = "meta-textgeneration-llama-2-7b"
+hub_name = "bencrab-test"
+
+model = JumpStartModel(
+    model_id=model_id,
+    hub_name=hub_name,
+)
+
+# model.deploy(accept_eula=True, endpoint_type=EndpointType.INFERENCE_COMPONENT_BASED)
+
+_retrieve_default_content_type(
+    model_id=model_id,
+    model_version="*",
+    region="us-west-2",
+    hub_arn=f"arn:aws:sagemaker:us-west-2:802376408542:hub/{hub_name}",
+)