Explorar el Código

to be compatible with ts inference yaml (#2682)

Tingquan Gao hace 11 meses
padre
commit
e05a4206e2
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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