change df serialization to avoid mem leak

This commit is contained in:
Timothy Pogue
2022-11-17 11:59:03 -07:00
parent 0a7f4fd3cc
commit 875e9ab447
3 changed files with 12 additions and 4 deletions

View File

@@ -3,7 +3,7 @@ from abc import ABC, abstractmethod
import orjson
import rapidjson
from pandas import DataFrame
from pandas import DataFrame, Timestamp
from freqtrade.misc import dataframe_to_json, json_to_dataframe
from freqtrade.rpc.api_server.ws.proxy import WebSocketProxy
@@ -52,6 +52,11 @@ def _json_default(z):
'__type__': 'dataframe',
'__value__': dataframe_to_json(z)
}
# Pandas returns a Timestamp object, we need to
# convert it to a timestamp int (with ms) for orjson
# to handle it
if isinstance(z, Timestamp):
return z.timestamp() * 1e3
raise TypeError