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

fix: fix a bug in scenario.py (#388)

Browse Source 
* fix: fix a bug in scenario.py

* Update scenario.py
This commit is contained in:
Haoran Pan
2024-09-29 14:46:00 +08:00
committed by GitHub
parent 26cc2b580d
commit 0467ba03a8
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/scenarios/kaggle/experiment/scenario.py
+2 -1
View File
@@ -37,6 +37,7 @@ class KGScenario(Scenario):
self.evaluation_desc = None
self.evaluation_metric_direction = None
self.vector_base = None
self.dsagent = KAGGLE_IMPLEMENT_SETTING.dsagent
self._analysis_competition_description()
self.if_action_choosing_based_on_UCB = KAGGLE_IMPLEMENT_SETTING.if_action_choosing_based_on_UCB
self.if_using_graph_rag = KAGGLE_IMPLEMENT_SETTING.if_using_graph_rag
@@ -44,7 +45,7 @@ class KGScenario(Scenario):
if self.if_using_vector_rag and KAGGLE_IMPLEMENT_SETTING.rag_path:
self.vector_base = KaggleExperienceBase(KAGGLE_IMPLEMENT_SETTING.rag_path)
self.vector_base.path = datetime.now(timezone.utc).strftime("%Y-%m-%d-%H-%M-%S") + "_kaggle_kb.pkl"
self.vector_base.path = Path(datetime.now(timezone.utc).strftime("%Y-%m-%d-%H-%M-%S") + "_kaggle_kb.pkl")
self.vector_base.dump()
self._output_format = self.output_format