Fix

cohere-ai · Oct 18, 2024 · ab6ae1b · ab6ae1b
1 parent ad0cd3d
commit ab6ae1b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/cohere/aws_client.py b/src/cohere/aws_client.py
@@ -19,7 +19,7 @@
     from botocore.auth import SigV4Auth  # type: ignore
     from botocore.awsrequest import AWSRequest  # type: ignore
     AWS_DEPS_AVAILABLE = True
- except ImportError:
+except ImportError:
     AWS_DEPS_AVAILABLE = False
 
 class AwsClient(Client):

diff --git a/src/cohere/manually_maintained/cohere_aws/client.py b/src/cohere/manually_maintained/cohere_aws/client.py
@@ -25,9 +25,10 @@
     from botocore.exceptions import (
         ClientError, EndpointConnectionError, ParamValidationError)
     AWS_DEPS_AVAILABLE = True
- except ImportError:
+except ImportError:
     AWS_DEPS_AVAILABLE = False
 
+class Client:
     def __init__(
            self,
             aws_region: typing.Optional[str] = None,
@@ -38,7 +39,7 @@ def __init__(
         """
         if not AWS_DEPS_AVAILABLE:
             raise CohereError("AWS dependencies not available. Please install boto3 and sagemaker.")
-        self._client = boto3.client(
+        self._client = boto3.client()
         if os.environ.get('AWS_DEFAULT_REGION') is None:
             os.environ['AWS_DEFAULT_REGION'] = aws_region
         self._sess = sage.Session(sagemaker_client=self._service_client)