Fixes

cohere-ai · Sep 30, 2024 · 1b71abe · 1b71abe
1 parent 33c8b91
commit 1b71abe
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)
+        self._sess = sage.Session(sagemaker_client=self._service_client, region_name=aws_region)
         self.mode = Mode.SAGEMAKER