ghndrx/prowler
1
0
Fork 0
mirror of https://github.com/ghndrx/prowler.git synced 2026-02-11 15:25:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(sagemaker): Handle ValidationException (#2321)

Browse Source
This commit is contained in:
Pepe Fagoaga
2023-05-08 14:52:28 +02:00
committed by GitHub
parent c930f08348
commit 8a02574303
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
prowler/providers/aws/services/sagemaker/sagemaker_service.py
View File

@@ -1,6 +1,7 @@
import threading
from typing import Optional
from botocore.client import ClientError
from pydantic import BaseModel
from prowler.lib.logger import logger
@@ -115,9 +116,19 @@ class SageMaker:
try:
for notebook_instance in self.sagemaker_notebook_instances:
regional_client = regional_clients[notebook_instance.region]
describe_notebook_instance = regional_client.describe_notebook_instance(
NotebookInstanceName=notebook_instance.name
)
try:
describe_notebook_instance = (
regional_client.describe_notebook_instance(
NotebookInstanceName=notebook_instance.name
)
)
except ClientError as error:
if error.response["Error"]["Code"] == "ValidationException":
logger.warning(
f"{regional_client.region} -- {error.__class__.__name__}[{error.__traceback__.tb_lineno}]: {error}"
)
continue
if (
"RootAccess" in describe_notebook_instance
and describe_notebook_instance["RootAccess"] == "Enabled"