gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-27 23:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: test kaggle method (#296)

Browse Source 
* fix test kaggle

* add hint for kaggle template test
This commit is contained in:
XianBW
2024-09-23 13:33:46 +08:00
committed by GitHub
parent e9629f9f11
commit d3db48e06e
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/utils/test_kaggle.py
+5 -2
View File
@@ -13,18 +13,21 @@ from rdagent.utils.agent.tpl import T
class TestTpl(unittest.TestCase):
def test_competition_template(self):
"""
export KG_COMPETITION=<competition_name> before running this test
"""
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"
/ f"{competition}_template"
)
print(ws.workspace_path)
ws.execute()
success = (ws.workspace_path / "submission.csv").exists()
self.assertTrue(success, "submission.csv is not generated")
ws.clear()
return success
if __name__ == "__main__":