mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 07:57:44 +00:00
Xu Yang
fce241b9f9
* Use ExtendedBaseSettings to replace BaseSettings * update a more general way to pass the default setting * update all code * fix CI * fix CI * fix qlib scenario * fix CI * fix CI * fix CI & add data science interfaces * remove redundant code * abandon costeer knowledge base v1 --------- Co-authored-by: Xu Yang <xuyang1@microsoft.com> Co-authored-by: XianBW <36835909+XianBW@users.noreply.github.com>
28 lines
990 B
Python
28 lines
990 B
Python
from pathlib import Path
|
|
|
|
import dill as pickle # type: ignore[import-untyped]
|
|
|
|
from rdagent.log import rdagent_logger as logger
|
|
|
|
|
|
class KnowledgeBase:
|
|
def __init__(self, path: str | Path | None = None) -> None:
|
|
self.path = Path(path) if path else None
|
|
self.load()
|
|
|
|
def load(self) -> None:
|
|
if self.path is not None and self.path.exists():
|
|
with self.path.open("rb") as f:
|
|
loaded = pickle.load(f)
|
|
if isinstance(loaded, dict):
|
|
self.__dict__.update({k: v for k, v in loaded.items() if k != "path"})
|
|
else:
|
|
self.__dict__.update({k: v for k, v in loaded.__dict__.items() if k != "path"})
|
|
|
|
def dump(self) -> None:
|
|
if self.path is not None:
|
|
self.path.parent.mkdir(parents=True, exist_ok=True)
|
|
pickle.dump(self.__dict__, self.path.open("wb"))
|
|
else:
|
|
logger.warning("KnowledgeBase path is not set, dump failed.")
|