Fixes

cohere-ai · Sep 30, 2024 · 83d55dc · 83d55dc
1 parent 89b8799
commit 83d55dc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/mypy.ini b/mypy.ini
@@ -0,0 +1,2 @@
+[mypy]
+exclude = src/cohere/manually_maintained/cohere_aws
diff --git a/src/cohere/sagemaker_client.py b/src/cohere/sagemaker_client.py
@@ -5,7 +5,7 @@
 
 
 class SagemakerClient(AwsClient):
-    finetuning: Client
+    sagemaker_finetuning: Client
 
     def __init__(
             self,
@@ -25,4 +25,4 @@ def __init__(
             aws_region=aws_region,
             timeout=timeout,
         )
-        self.finetuning = Client(region_name=self._aws_region)
+        self.sagemaker_finetuning = Client(region_name=aws_region)