Open yaml files with utf-8 encoding for extra_model_paths.yaml (#6807)

* Using utf-8 encoding for yaml files. * Fix test assertion.
comfyanonymous · Feb 14, 2025 · 042a905 · 042a905
1 parent 019c702
commit 042a905
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests-unit/utils/extra_config_test.py b/tests-unit/utils/extra_config_test.py
@@ -114,7 +114,7 @@ def test_load_extra_model_paths_expands_userpath(
     mock_yaml_safe_load.assert_called_once()
 
     # Check if open was called with the correct file path
-    mock_file.assert_called_once_with(dummy_yaml_file_name, 'r')
+    mock_file.assert_called_once_with(dummy_yaml_file_name, 'r', encoding='utf-8')
 
 
 @patch('builtins.open', new_callable=mock_open)

diff --git a/utils/extra_config.py b/utils/extra_config.py
@@ -4,7 +4,7 @@
 import logging
 
 def load_extra_path_config(yaml_path):
-    with open(yaml_path, 'r') as stream:
+    with open(yaml_path, 'r', encoding='utf-8') as stream:
         config = yaml.safe_load(stream)
     yaml_dir = os.path.dirname(os.path.abspath(yaml_path))
     for c in config: