Skip to content

Commit 4e8bcab

Browse files
nmadanNamrata Madan
authored andcommitted
fix: yaml safe_load sagemaker config
Co-authored-by: Namrata Madan <[email protected]>
1 parent f9560fc commit 4e8bcab

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/local/local_session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -667,7 +667,7 @@ def _initialize(
667667
logger.error(_module_import_error("yaml", "Local mode", "local"))
668668
raise e
669669

670-
self.config = yaml.load(open(sagemaker_config_file, "r"))
670+
self.config = yaml.safe_load(open(sagemaker_config_file, "r"))
671671
if self._disable_local_code and "local" in self.config:
672672
self.config["local"]["local_code"] = False
673673

0 commit comments

Comments
 (0)