feat: kaggle templates related (#287)

* add kaggle test

* kaggle templates changes
This commit is contained in:
XianBW
2024-09-20 20:49:44 +08:00
committed by GitHub
parent e152c68c88
commit f4dd0ffcd6
19 changed files with 781 additions and 41 deletions
+31
View File
@@ -0,0 +1,31 @@
import unittest
from pathlib import Path
import nbformat
from rich import print
from rdagent.app.kaggle.conf import KAGGLE_IMPLEMENT_SETTING
from rdagent.oai.llm_utils import APIBackend
from rdagent.scenarios.kaggle.experiment.workspace import KGFBWorkspace
from rdagent.utils.agent.ret import PythonAgentOut
from rdagent.utils.agent.tpl import T
class TestTpl(unittest.TestCase):
def test_competition_template(self):
competition = KAGGLE_IMPLEMENT_SETTING.competition
print(f"[bold orange]{competition}[/bold orange]")
ws = KGFBWorkspace(
template_folder_path=Path(__file__).parent.parent.parent
/ "rdagent/scenarios/kaggle/experiment"
/ f"{KAGGLE_IMPLEMENT_SETTING.competition}_template"
)
print(ws.workspace_path)
ws.execute()
success = (ws.workspace_path / "submission.csv").exists()
ws.clear()
return success
if __name__ == "__main__":
unittest.main()