mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 16:07:46 +00:00
you-n-g
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
"""
|
|
CLI entrance for all rdagent application.
|
|
|
|
This will
|
|
- make rdagent a nice entry and
|
|
- autoamtically load dotenv
|
|
"""
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv(".env")
|
|
# 1) Make sure it is at the beginning of the script so that it will load dotenv before initializing BaseSettings.
|
|
# 2) The ".env" argument is necessary to make sure it loads `.env` from the current directory.
|
|
|
|
import subprocess
|
|
from importlib.resources import path as rpath
|
|
|
|
import fire
|
|
|
|
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
|
|
from rdagent.app.utils.info import collect_info
|
|
|
|
|
|
def ui(port=80, log_dir="", debug=False):
|
|
"""
|
|
start web app to show the log traces.
|
|
"""
|
|
with rpath("rdagent.log.ui", "app.py") as app_path:
|
|
cmds = ["streamlit", "run", app_path, f"--server.port={port}"]
|
|
if log_dir or debug:
|
|
cmds.append("--")
|
|
if log_dir:
|
|
cmds.append(f"--log_dir={log_dir}")
|
|
if debug:
|
|
cmds.append("--debug")
|
|
subprocess.run(cmds)
|
|
|
|
|
|
def app():
|
|
fire.Fire(
|
|
{
|
|
"fin_factor": fin_factor,
|
|
"fin_factor_report": fin_factor_report,
|
|
"fin_model": fin_model,
|
|
"med_model": med_model,
|
|
"general_model": general_model,
|
|
"ui": ui,
|
|
"collect_info": collect_info,
|
|
"kaggle": kaggle_main,
|
|
}
|
|
)
|