Merge pull request #248 from leeeizhang/lei/quickfix-kaggle

[MRG] quickfix: miss list_files function
MLSysOps · Oct 16, 2024 · 4018824 · 4018824
2 parents 699f858 + e8d1130
commit 4018824
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mle/agents/advisor.py b/mle/agents/advisor.py
@@ -126,7 +126,8 @@ def __init__(self, model, console=None, mode='normal'):
             schema_web_search,
             schema_search_arxiv,
             schema_search_papers_with_code,
-            schema_preview_csv_data
+            schema_preview_csv_data,
+            schema_list_files
         ]
         if config_data.get('search_key'):
             self.functions.append(schema_web_search)