Skip to content

Commit 72e7c3b

Browse files
authored
feat: add kaggle command (microsoft#271)
* feat: add kaggle command * lint
1 parent 204580b commit 72e7c3b

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

rdagent/app/cli.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
from rdagent.app.general_model.general_model import (
2121
extract_models_and_implement as general_model,
2222
)
23+
from rdagent.app.kaggle.loop import main as kaggle_main
2324
from rdagent.app.qlib_rd_loop.factor import main as fin_factor
2425
from rdagent.app.qlib_rd_loop.factor_from_report import main as fin_factor_report
2526
from rdagent.app.qlib_rd_loop.model import main as fin_model
@@ -51,5 +52,6 @@ def app():
5152
"general_model": general_model,
5253
"ui": ui,
5354
"collect_info": collect_info,
55+
"kaggle": kaggle_main,
5456
}
5557
)

rdagent/app/kaggle/loop.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -77,9 +77,10 @@ def main(path=None, step_n=None, competition=None):
7777
7878
You can continue running session by
7979
80-
.. code-block:: python
80+
.. code-block:: bash
8181
8282
dotenv run -- python rdagent/app/kaggle/loop.py [--competition titanic] $LOG_PATH/__session__/1/0_propose --step_n 1 # `step_n` is a optional paramter
83+
rdagent kaggle --competition playground-series-s4e8 # You are encouraged to use this one.
8384
8485
"""
8586
if competition:
@@ -92,7 +93,4 @@ def main(path=None, step_n=None, competition=None):
9293

9394

9495
if __name__ == "__main__":
95-
from dotenv import load_dotenv
96-
97-
load_dotenv(override=True)
9896
fire.Fire(main)

0 commit comments

Comments
 (0)