fix: boto_session

evakravi · evakravi · commit 254f147a581f · 2024-01-31T23:20:28.000Z
diff --git a/tests/integ/sagemaker/jumpstart/conftest.py b/tests/integ/sagemaker/jumpstart/conftest.py
@@ -50,7 +50,7 @@ def _teardown():
         config=Config(retries={"max_attempts": 10, "mode": "standard"}),
     )
 
-    sagemaker_session = Session(boto3_session=boto3_session, sagemaker_client=sagemaker_client)
+    sagemaker_session = Session(boto_session=boto3_session, sagemaker_client=sagemaker_client)
 
     search_endpoints_result = sagemaker_client.search(
         Resource="Endpoint",

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ def _teardown():`
`50`	`50`	`config=Config(retries={"max_attempts": 10, "mode": "standard"}),`
`51`	`51`	`)`
`52`	`52`
`53`		`- sagemaker_session = Session(boto3_session=boto3_session, sagemaker_client=sagemaker_client)`
	`53`	`+ sagemaker_session = Session(boto_session=boto3_session, sagemaker_client=sagemaker_client)`
`54`	`54`
`55`	`55`	`search_endpoints_result = sagemaker_client.search(`
`56`	`56`	`Resource="Endpoint",`