fix: JumpStartModelSpecs hyperparameters

evakravi · evakravi · commit 02b1a971c9bc · 2021-11-18T21:33:40.000Z
diff --git a/src/sagemaker/jumpstart/types.py b/src/sagemaker/jumpstart/types.py
@@ -173,7 +173,7 @@ def from_json(self, json_obj: Dict[str, Any]) -> None:
             )
             self.training_artifact_key: Optional[str] = json_obj["training_artifact_key"]
             self.training_script_key: Optional[str] = json_obj["training_script_key"]
-            self.hyperparameters: Optional[Dict[str, Any]] = json_obj["hyperparameters"]
+            self.hyperparameters: Optional[Dict[str, Any]] = json_obj.get("hyperparameters")
         else:
             self.training_ecr_specs = (
                 self.training_artifact_key

Original file line number	Diff line number	Diff line change
`@@ -173,7 +173,7 @@ def from_json(self, json_obj: Dict[str, Any]) -> None:`
`173`	`173`	`)`
`174`	`174`	`self.training_artifact_key: Optional[str] = json_obj["training_artifact_key"]`
`175`	`175`	`self.training_script_key: Optional[str] = json_obj["training_script_key"]`
`176`		`- self.hyperparameters: Optional[Dict[str, Any]] = json_obj["hyperparameters"]`
	`176`	`+ self.hyperparameters: Optional[Dict[str, Any]] = json_obj.get("hyperparameters")`
`177`	`177`	`else:`
`178`	`178`	`self.training_ecr_specs = (`
`179`	`179`	`self.training_artifact_key`