From 50aa0909ce2b51fc470f338daea5eff6b55689ca Mon Sep 17 00:00:00 2001
From: Billy Trend <billy@cohere.com>
Date: Mon, 30 Sep 2024 11:45:38 +0100
Subject: [PATCH] Fix region

---
 tests/test_aws_client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_aws_client.py b/tests/test_aws_client.py
index a6a60b4df..81789a679 100644
--- a/tests/test_aws_client.py
+++ b/tests/test_aws_client.py
@@ -40,7 +40,7 @@
         "platform": "sagemaker",
         "client": cohere.SagemakerClient(
             timeout=10000,
-            aws_region="us-east-1",
+            aws_region="us-east-2",
             aws_access_key="...",
             aws_secret_key="...",
             aws_session_token="...",