Skip to content

Commit cc8a500

Browse files
new approach for vulnerability
1 parent 82c4457 commit cc8a500

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

src/sagemaker/config/config_utils.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -68,13 +68,15 @@ def _log_sagemaker_config_single_substitution(source_value, config_value, config
6868
"""
6969
logger = get_sagemaker_config_logger()
7070

71-
sensitive_keywords = re.compile(r'(secret|password|key)', re.IGNORECASE)
71+
# sensitive_keywords = re.compile(r'(secret|password|key)', re.IGNORECASE)
7272

73-
if sensitive_keywords.search(config_key_path):
74-
if source_value is not None:
75-
source_value = "***"
76-
if config_value is not None:
77-
config_value = "***"
73+
if isinstance(source_value, dict):
74+
for key in source_value.keys():
75+
source_value[key] = '***'
76+
77+
if isinstance(config_value, dict):
78+
for key in config_value.keys():
79+
config_value[key] = '***'
7880

7981
if config_value is not None:
8082

0 commit comments

Comments
 (0)