CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2026-01-29 02:10:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: move joblib_writer to use file interface

Browse Source
This commit is contained in:
Matthias
2024-12-16 19:14:46 +01:00
parent 2a9a3e020c
commit 37bbb0e0e0
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
freqtrade/optimize/optimize_reports/bt_storage.py
View File

@@ -1,6 +1,6 @@
import logging
from pathlib import Path
from typing import Any
from typing import Any, TextIO
from pandas import DataFrame
@@ -14,7 +14,7 @@ from freqtrade.optimize.backtest_caching import get_backtest_metadata_filename
logger = logging.getLogger(__name__)
def file_dump_joblib(filename: Path, data: Any, log: bool = True) -> None:
def file_dump_joblib(file_obj: TextIO, data: Any, log: bool = True) -> None:
"""
Dump object data into a file
:param filename: file to create
@@ -23,11 +23,7 @@ def file_dump_joblib(filename: Path, data: Any, log: bool = True) -> None:
"""
import joblib
if log:
logger.info(f'dumping joblib to "{filename}"')
with filename.open("wb") as fp:
joblib.dump(data, fp)
logger.debug(f'done joblib dump to "{filename}"')
joblib.dump(data, file_obj)
def _generate_filename(recordfilename: Path, appendix: str, suffix: str) -> Path:
@@ -122,6 +118,9 @@ def _store_backtest_analysis_data(
"""
filename = _generate_filename(recordfilename, f"{dtappendix}_{name}", ".pkl")
file_dump_joblib(filename, data)
logger.info(f'dumping joblib to "{filename}"')
with filename.open("wb") as fp:
file_dump_joblib(fp, data)
logger.debug(f'done joblib dump to "{filename}"')
return filename