From 05e38410e9e962b66795951cebf58cabc442067e Mon Sep 17 00:00:00 2001 From: Billy Trend Date: Mon, 30 Sep 2024 12:07:07 +0100 Subject: [PATCH] Sesh --- src/cohere/manually_maintained/cohere_aws/client.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cohere/manually_maintained/cohere_aws/client.py b/src/cohere/manually_maintained/cohere_aws/client.py index 8cd355576..31a9b2ec8 100644 --- a/src/cohere/manually_maintained/cohere_aws/client.py +++ b/src/cohere/manually_maintained/cohere_aws/client.py @@ -34,7 +34,7 @@ def __init__( """ self._client = boto3.client("sagemaker-runtime", region_name=aws_region) self._service_client = boto3.client("sagemaker", region_name=aws_region) - self._sess = sage.Session(sagemaker_client=self._service_client, region_name=aws_region) + self._sess = sage.Session(boto3.session.Session()) self.mode = Mode.SAGEMAKER