Fix failing integ test

claytonparnell · claytonparnell · commit 93671a51805d · 2023-05-30T10:58:46.000-07:00
diff --git a/tests/integ/test_sagemaker_config.py b/tests/integ/test_sagemaker_config.py
@@ -344,6 +344,7 @@ def test_sagemaker_config_cross_context_injection(
             "InferenceExecutionConfig": {"Mode": "Serial"},
             "ExecutionRoleArn": role_arn,  # from sagemaker_config
             "EnableNetworkIsolation": True,  # from sagemaker_config
+            "DeploymentRecommendation": {"RecommendationStatus": "NOT_APPLICABLE"},
         }
 
         expected_endpoint_1 = {

Original file line number	Diff line number	Diff line change
`@@ -344,6 +344,7 @@ def test_sagemaker_config_cross_context_injection(`
`344`	`344`	`"InferenceExecutionConfig": {"Mode": "Serial"},`
`345`	`345`	`"ExecutionRoleArn": role_arn, # from sagemaker_config`
`346`	`346`	`"EnableNetworkIsolation": True, # from sagemaker_config`
	`347`	`+ "DeploymentRecommendation": {"RecommendationStatus": "NOT_APPLICABLE"},`
`347`	`348`	`}`
`348`	`349`
`349`	`350`	`expected_endpoint_1 = {`