From ab6ae1b733c85237ce5cfe201c92a9a35c0cc1d2 Mon Sep 17 00:00:00 2001 From: Billy Trend Date: Fri, 18 Oct 2024 17:17:56 +0100 Subject: [PATCH] Fix --- src/cohere/aws_client.py | 2 +- src/cohere/manually_maintained/cohere_aws/client.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/cohere/aws_client.py b/src/cohere/aws_client.py index 95099682f..f34d60a54 100644 --- 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 index 5d2a852c4..60addcc7d 100644 --- 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)