Revert to FullLoader with context manager for model YAML with Python tags

keybangz · keybangz · commit 328cc268e7e8 · 2026-03-17T21:46:51.000+13:00
diff --git a/audio_separator/separator/separator.py b/audio_separator/separator/separator.py
@@ -1058,7 +1058,7 @@ def load_model_data_from_yaml(self, yaml_config_filename):
         )
 
         with open(model_data_yaml_filepath, encoding="utf-8") as f:
-            model_data = yaml.safe_load(f)
+            model_data = yaml.load(f, Loader=yaml.FullLoader)
         self.logger.debug(f"Model data loaded from YAML file: {model_data}")
 
         if "roformer" in model_data_yaml_filepath.lower():

Original file line number	Diff line number	Diff line change
`@@ -1058,7 +1058,7 @@ def load_model_data_from_yaml(self, yaml_config_filename):`
`1058`	`1058`	`)`
`1059`	`1059`
`1060`	`1060`	`with open(model_data_yaml_filepath, encoding="utf-8") as f:`
`1061`		`- model_data = yaml.safe_load(f)`
	`1061`	`+ model_data = yaml.load(f, Loader=yaml.FullLoader)`
`1062`	`1062`	`self.logger.debug(f"Model data loaded from YAML file: {model_data}")`
`1063`	`1063`
`1064`	`1064`	`if "roformer" in model_data_yaml_filepath.lower():`