Explorar o código

fix saving pd.DataFrame to json

gaotingquan hai 1 ano
pai
achega
4fcfdd135e
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      paddlex/inference/results/utils/mixin.py

+ 1 - 1
paddlex/inference/results/utils/mixin.py

@@ -79,7 +79,7 @@ class JsonMixin:
             elif isinstance(obj, np.ndarray):
                 return [_format_data(item) for item in obj.tolist()]
             elif isinstance(obj, pd.DataFrame):
-                return obj.to_json(orient="records", force_ascii=False)
+                return json.loads(obj.to_json(orient="records", force_ascii=False))
             elif isinstance(obj, Path):
                 return obj.as_posix()
             elif isinstance(obj, dict):