瀏覽代碼

to be compatible with ts inference yaml (#2682)

Tingquan Gao 11 月之前
父節點
當前提交
e05a4206e2
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/inference/utils/io/readers.py

+ 1 - 1
paddlex/inference/utils/io/readers.py

@@ -349,5 +349,5 @@ class YAMLReaderBackend(_BaseReaderBackend):
 
     def read_file(self, in_path, **kwargs):
         with open(in_path, "r", encoding="utf-8", **kwargs) as yaml_file:
-            data = yaml.safe_load(yaml_file)
+            data = yaml.load(yaml_file, Loader=yaml.FullLoader)
         return data