Sesh

cohere-ai · Sep 30, 2024 · 05e3841 · 05e3841
1 parent 1b71abe
commit 05e3841
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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