Prechádzať zdrojové kódy

fix saving pd.DataFrame to json (#2870)

Tingquan Gao 10 mesiacov pred
rodič
commit
9248f27643

+ 1 - 2
paddlex/inference/common/result/base_ts_result.py

@@ -35,8 +35,7 @@ class BaseTSResult(BaseResult, CSVMixin):
         assert (
             BaseTSResult.INPUT_TS_KEY in data
         ), f"`{BaseTSResult.INPUT_TS_KEY}` is needed, but not found in `{list(data.keys())}`!"
-        self._input_ts = data.pop("input_ts", None)
-        self._ts_writer = CSVWriter(backend="pandas")
+        data.pop("input_ts", None)
 
         super().__init__(data)
         CSVMixin.__init__(self, "pandas")

+ 1 - 1
paddlex/inference/common/result/mixin.py

@@ -96,7 +96,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):