fix: sagemaker session to delete IC

evakravi · evakravi · commit 13b7b2cd0cf7 · 2024-02-01T03:36:22.000Z
diff --git a/tests/integ/sagemaker/jumpstart/conftest.py b/tests/integ/sagemaker/jumpstart/conftest.py
@@ -92,8 +92,8 @@ def _teardown():
 
     # delete inference components for test-suite-tagged endpoints
     for inference_component_name in inference_component_names:
-        sagemaker_client.delete_inference_component(
-            InferenceComponentName=inference_component_name, wait=True
+        sagemaker_session.delete_inference_component(
+            inference_component_name=inference_component_name, wait=True
         )
 
     # delete test-suite-tagged endpoints

Original file line number	Diff line number	Diff line change
`@@ -92,8 +92,8 @@ def _teardown():`
`92`	`92`
`93`	`93`	`# delete inference components for test-suite-tagged endpoints`
`94`	`94`	`for inference_component_name in inference_component_names:`
`95`		`- sagemaker_client.delete_inference_component(`
`96`		`- InferenceComponentName=inference_component_name, wait=True`
	`95`	`+ sagemaker_session.delete_inference_component(`
	`96`	`+ inference_component_name=inference_component_name, wait=True`
`97`	`97`	`)`
`98`	`98`
`99`	`99`	`# delete test-suite-tagged endpoints`