fix: use as_unit instead of int math

pull/13063/head
Matthias 1 week ago
parent 187d06b5ba
commit 74ba9d76a2

@ -11,7 +11,7 @@ from typing import TYPE_CHECKING, Any
import psutil
from dateutil.relativedelta import relativedelta
from dateutil.tz import tzlocal
from numpy import inf, int64, isnan, mean, nan
from numpy import inf, isnan, mean, nan
from pandas import DataFrame, NaT, read_sql
from sqlalchemy import func, select
@ -1536,7 +1536,9 @@ class RPC:
df_cols = [col for col in dataframe_columns if col in cols_set]
dataframe = dataframe.loc[:, df_cols]
dataframe.loc[:, "__date_ts"] = dataframe.loc[:, "date"].dt.as_unit("ms").astype(int64)
dataframe.loc[:, "__date_ts"] = (
dataframe.loc[:, "date"].dt.as_unit("ms").astype("int64")
)
# Move signal close to separate column when signal for easy plotting
for sig_type in signals.keys():
if sig_type in dataframe.columns:

@ -207,7 +207,7 @@ def generate_test_data(
def generate_test_data_raw(timeframe: str, size: int, start: str = "2020-07-05", random_seed=42):
"""Generates data in the ohlcv format used by ccxt"""
df = generate_test_data(timeframe, size, start, random_seed)
df["date"] = df.loc[:, "date"].dt.as_unit("ms").astype(np.int64)
df["date"] = df.loc[:, "date"].dt.as_unit("ms").astype("int64")
return list(list(x) for x in zip(*(df[x].values.tolist() for x in df.columns), strict=False))

@ -69,7 +69,7 @@ def make_response_from_url(start_date, end_date):
"taker_buy_quote_volume,ignore"
)
df = pd.DataFrame(columns=cols.split(","), dtype=float)
df["open_time"] = date_col.astype("int64") // 10**6
df["open_time"] = date_col.as_unit("ms").astype("int64")
df["open"] = df["high"] = df["low"] = df["close"] = df["volume"] = 1.0
return df

Loading…
Cancel
Save