fix: docstring, add integ test

Author: evakravi

src/sagemaker/payloads.py

@@ -45,8 +45,8 @@ def retrieve_samples(
         model_version (str): The version of the JumpStart model for which to retrieve
             the model payloads.
         serialize (bool): Whether to serialize byte-stream valued payloads by downloading
-            binary files from s3 and applying encoding, or to keep payload in pre-serialized .
-            state. Set this option to False if you only want to avoid s3 download of it you
+            binary files from s3 and applying encoding, or to keep payload in pre-serialized
+            state. Set this option to False if you want to avoid s3 downloads or if you
             want to inspect the payload in a human-readable form. (Default: True).
         tolerate_vulnerable_model (bool): ``True`` if vulnerable versions of model
             specifications should be tolerated without raising an exception. If ``False``, raises an

tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py

@@ -86,6 +86,31 @@ def test_prepacked_jumpstart_model(setup):
     assert response is not None
 
 
+def test_default_payload_jumpstart_model(setup):
+
+    ## DO NOT COMMIT THIS LINE
+    os.environ.update({"AWS_JUMPSTART_CONTENT_BUCKET_OVERRIDE": "jumpstart-cache-alpha-us-west-2"})
+
+    model_id = "model-depth2img-stable-diffusion-v1-5-controlnet-v1-1-fp16"
+
+    model = JumpStartModel(
+        model_id=model_id,
+        role=get_sm_session().get_caller_identity_arn(),
+        sagemaker_session=get_sm_session(),
+    )
+
+    default_payload = model.retrieve_default_payload()
+
+    # uses ml.g5.8xlarge instance
+    predictor = model.deploy(
+        tags=[{"Key": JUMPSTART_TAG, "Value": os.environ[ENV_VAR_JUMPSTART_SDK_TEST_SUITE_ID]}],
+    )
+
+    response = predictor.predict(default_payload)
+
+    assert response is not None
+
+
 @pytest.mark.skipif(
     tests.integ.test_region() not in GATED_INFERENCE_MODEL_SUPPORTED_REGIONS,
     reason=f"JumpStart gated inference models unavailable in {tests.integ.test_region()}.",