fix: script scope should be training

Author: evakravi

src/sagemaker/jumpstart/artifacts.py

@@ -292,12 +292,12 @@ def _retrieve_script_uri(
     )
 
     if script_scope == JumpStartScriptScope.INFERENCE:
-        model_script_key = (
-            getattr(model_specs, "training_prepacked_script_key") or model_specs.hosting_script_key
-        )
+        model_script_key = model_specs.hosting_script_key
 
     elif script_scope == JumpStartScriptScope.TRAINING:
-        model_script_key = model_specs.training_script_key
+        model_script_key = (
+            getattr(model_specs, "training_prepacked_script_key") or model_specs.training_script_key
+        )
 
     bucket = os.environ.get(
         ENV_VARIABLE_JUMPSTART_SCRIPT_ARTIFACT_BUCKET_OVERRIDE

tests/unit/sagemaker/script_uris/jumpstart/test_combined_script_artifact.py

@@ -28,7 +28,7 @@ def test_jumpstart_combined_artifacts(patched_get_model_specs):
 
     uri = script_uris.retrieve(
         region="us-west-2",
-        script_scope="inference",
+        script_scope="training",
         model_id=model_id_combined_script_artifact,
         model_version="*",
     )