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

* fix: fix a bug in scenario.py * Update scenario.py
microsoft · Sep 29, 2024 · 999a1eb · 999a1eb
1 parent 8fe9511
commit 999a1eb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/rdagent/scenarios/kaggle/experiment/scenario.py b/rdagent/scenarios/kaggle/experiment/scenario.py
@@ -37,14 +37,15 @@ 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
         self.if_using_vector_rag = KAGGLE_IMPLEMENT_SETTING.if_using_vector_rag
 
         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