diff --git a/backend/deepchecks_monitoring/ee/api/v1/data_sources.py b/backend/deepchecks_monitoring/ee/api/v1/data_sources.py index d43153b1d..6a49227e6 100644 --- a/backend/deepchecks_monitoring/ee/api/v1/data_sources.py +++ b/backend/deepchecks_monitoring/ee/api/v1/data_sources.py @@ -13,7 +13,7 @@ import boto3 import sqlalchemy as sa from botocore.config import Config -from botocore.exceptions import BotoCoreError +from botocore.exceptions import ClientError from fastapi import Depends, Response from pydantic.main import BaseModel from sqlalchemy.ext.asyncio import AsyncSession @@ -74,7 +74,7 @@ async def new_data_source(body: DataSourceCreationSchema, config=Config(region_name=data_source.parameters['region']) ) sts.get_caller_identity() - except BotoCoreError as e: + except ClientError as e: raise BadRequest('Invalid credentials to AWS') from e else: raise BadRequest('Invalid data source type')