From 5f9ea6b37c5d4fd31f810bde6f325fa28efa8a66 Mon Sep 17 00:00:00 2001 From: shayts7 Date: Mon, 12 Jun 2023 15:43:34 +0300 Subject: [PATCH] Switched to ClientError exception --- backend/deepchecks_monitoring/ee/api/v1/data_sources.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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')