fix: fix a bug in kaggle conf (#459)

* fix a bug in kaggle conf * fix a ci error
microsoft · Oct 29, 2024 · b4ed32b · b4ed32b
1 parent 58cf397
commit b4ed32b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/rdagent/app/kaggle/conf.py b/rdagent/app/kaggle/conf.py
@@ -63,9 +63,7 @@ class Config:
     """Enable advanced graph-based RAG"""
 
     # Conditionally set the knowledge_base based on the use of graph RAG
-    knowledge_base: str = (
-        "rdagent.scenarios.kaggle.knowledge_management.graph.KGKnowledgeGraph" if if_using_graph_rag else ""
-    )
+    knowledge_base: str = ""
     """Knowledge base class, uses 'KGKnowledgeGraph' when advanced graph-based RAG is enabled, otherwise empty."""
 
     knowledge_base_path: str = "kg_graph.pkl"

diff --git a/rdagent/app/kaggle/loop.py b/rdagent/app/kaggle/loop.py
@@ -129,6 +129,10 @@ def main(path=None, step_n=None, competition=None):
     if competition:
         KAGGLE_IMPLEMENT_SETTING.competition = competition
         download_data(competition=competition, local_path=KAGGLE_IMPLEMENT_SETTING.local_data_path)
+        if KAGGLE_IMPLEMENT_SETTING.if_using_graph_rag:
+            KAGGLE_IMPLEMENT_SETTING.knowledge_base = (
+                "rdagent.scenarios.kaggle.knowledge_management.graph.KGKnowledgeGraph"
+            )
     else:
         logger.error("Please specify competition name.")
     if path is None: