fix: Unable to attach estimator to training job when KeepAlivePeriodInSeconds specified (#3674)

TheOnly92 · web-flow · commit 4ca7d419a139 · 2023-05-02T10:31:35.000-07:00
diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
@@ -1702,7 +1702,7 @@ def _prepare_init_params_from_job_description(cls, job_details, model_channel_na
 
         if "KeepAlivePeriodInSeconds" in job_details["ResourceConfig"]:
             init_params["keep_alive_period_in_seconds"] = job_details["ResourceConfig"][
-                "keepAlivePeriodInSeconds"
+                "KeepAlivePeriodInSeconds"
             ]
 
         has_hps = "HyperParameters" in job_details

Original file line number	Diff line number	Diff line change
`@@ -1702,7 +1702,7 @@ def _prepare_init_params_from_job_description(cls, job_details, model_channel_na`
`1702`	`1702`
`1703`	`1703`	`if "KeepAlivePeriodInSeconds" in job_details["ResourceConfig"]:`
`1704`	`1704`	`init_params["keep_alive_period_in_seconds"] = job_details["ResourceConfig"][`
`1705`		`- "keepAlivePeriodInSeconds"`
	`1705`	`+ "KeepAlivePeriodInSeconds"`
`1706`	`1706`	`]`
`1707`	`1707`
`1708`	`1708`	`has_hps = "HyperParameters" in job_details`