From 999a1eb0eff9088e1b02419db741db4acf8d9ff7 Mon Sep 17 00:00:00 2001 From: Haoran Pan <167847254+TPLin22@users.noreply.github.com> Date: Sun, 29 Sep 2024 14:46:00 +0800 Subject: [PATCH] fix: fix a bug in scenario.py (#388) * fix: fix a bug in scenario.py * Update scenario.py --- rdagent/scenarios/kaggle/experiment/scenario.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rdagent/scenarios/kaggle/experiment/scenario.py b/rdagent/scenarios/kaggle/experiment/scenario.py index 67ffd93e..4023caf0 100644 --- a/rdagent/scenarios/kaggle/experiment/scenario.py +++ b/rdagent/scenarios/kaggle/experiment/scenario.py @@ -37,6 +37,7 @@ def __init__(self, competition: str) -> None: 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 @@ def __init__(self, competition: str) -> None: 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