Fix JumpStartModel.AdditionalModelDataSource model access config assignment.

Joseph Zhang · Joseph Zhang · commit 8fb27a0458ff · 2024-11-11T17:20:57.000-08:00
diff --git a/src/sagemaker/jumpstart/factory/model.py b/src/sagemaker/jumpstart/factory/model.py
@@ -588,7 +588,9 @@ def _apply_accept_eula_on_model_data_source(
             )
         )
 
-    mutable_model_data_source["model_access_config"] = {"accept_eula": accept_eula}
+    mutable_model_data_source["s3_data_source"]["model_access_config"] = {
+        "accept_eula": accept_eula
+    }
     return mutable_model_data_source
 
 

Original file line number	Diff line number	Diff line change
`@@ -588,7 +588,9 @@ def _apply_accept_eula_on_model_data_source(`
`588`	`588`	`)`
`589`	`589`	`)`
`590`	`590`
`591`		`- mutable_model_data_source["model_access_config"] = {"accept_eula": accept_eula}`
	`591`	`+ mutable_model_data_source["s3_data_source"]["model_access_config"] = {`
	`592`	`+ "accept_eula": accept_eula`
	`593`	`+ }`
`592`	`594`	`return mutable_model_data_source`
`593`	`595`
`594`	`596`