diff --git a/rdagent/app/cli.py b/rdagent/app/cli.py index 0f3411a3..0038599d 100644 --- a/rdagent/app/cli.py +++ b/rdagent/app/cli.py @@ -20,6 +20,7 @@ from rdagent.app.data_mining.model import main as med_model from rdagent.app.general_model.general_model import ( extract_models_and_implement as general_model, ) +from rdagent.app.kaggle.loop import main as kaggle_main from rdagent.app.qlib_rd_loop.factor import main as fin_factor from rdagent.app.qlib_rd_loop.factor_from_report import main as fin_factor_report from rdagent.app.qlib_rd_loop.model import main as fin_model @@ -51,5 +52,6 @@ def app(): "general_model": general_model, "ui": ui, "collect_info": collect_info, + "kaggle": kaggle_main, } ) diff --git a/rdagent/app/kaggle/loop.py b/rdagent/app/kaggle/loop.py index 0e823a71..96fb5fe7 100644 --- a/rdagent/app/kaggle/loop.py +++ b/rdagent/app/kaggle/loop.py @@ -77,9 +77,10 @@ def main(path=None, step_n=None, competition=None): You can continue running session by - .. code-block:: python + .. code-block:: bash 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 + rdagent kaggle --competition playground-series-s4e8 # You are encouraged to use this one. """ if competition: @@ -92,7 +93,4 @@ def main(path=None, step_n=None, competition=None): if __name__ == "__main__": - from dotenv import load_dotenv - - load_dotenv(override=True) fire.Fire(main)