gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-01 17:37:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

when restart with kb and different pkl path, we need to initialize the kb from json file (#835)

Browse Source
This commit is contained in:
Xu Yang
2025-04-28 01:51:51 -06:00
committed by GitHub
parent 6be059dd5d
commit 7cbf5bf56c
1 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/app/data_science/loop.py
+4 -7
View File
@@ -228,13 +228,10 @@ class DataScienceRDLoop(RDLoop):
replace_timer: bool = True,
) -> "LoopBase":
session = super().load(path, output_path, do_truncate, replace_timer)
if (
DS_RD_SETTING.enable_knowledge_base
and DS_RD_SETTING.knowledge_base_version == "v1"
and Path(DS_RD_SETTING.knowledge_base_path).exists()
):
knowledge_base = DSKnowledgeBase(path=DS_RD_SETTING.knowledge_base_path)
session.trace.knowledge_base = knowledge_base
if DS_RD_SETTING.enable_knowledge_base and DS_RD_SETTING.knowledge_base_version == "v1":
session.trace.knowledge_base = DSKnowledgeBase(
path=DS_RD_SETTING.knowledge_base_path, idea_pool_json_path=DS_RD_SETTING.idea_pool_json_path
)
return session
def dump(self, path: str | Path) -> None: