fix: Fix failing integ test (#3889)

claytonparnell · web-flow · commit 4903d9bbabe6 · 2023-05-31T09:24:04.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 = {`