mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 16:07:46 +00:00
Roland Minrui
f3cb3f8870
* add prev loops to runner history * fix evolving history * fix bug on initializing feedback without final decision * reformat * refine * add comments * fix ci * a little refinement * fix CI --------- Co-authored-by: Xu <v-xuminrui@microsoft.com> Co-authored-by: Xu Yang <peteryang@vip.qq.com>
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
import os
|
|
from typing import Optional
|
|
|
|
from pydantic_settings import SettingsConfigDict
|
|
|
|
from rdagent.components.coder.CoSTEER.config import CoSTEERSettings
|
|
from rdagent.utils.env import CondaConf, Env, LocalEnv
|
|
|
|
|
|
class FactorCoSTEERSettings(CoSTEERSettings):
|
|
model_config = SettingsConfigDict(env_prefix="FACTOR_CoSTEER_")
|
|
|
|
data_folder: str = "git_ignore_folder/factor_implementation_source_data"
|
|
"""Path to the folder containing financial data (default is fundamental data in Qlib)"""
|
|
|
|
data_folder_debug: str = "git_ignore_folder/factor_implementation_source_data_debug"
|
|
"""Path to the folder containing partial financial data (for debugging)"""
|
|
|
|
simple_background: bool = False
|
|
"""Whether to use simple background information for code feedback"""
|
|
|
|
file_based_execution_timeout: int = 3600
|
|
"""Timeout in seconds for each factor implementation execution"""
|
|
|
|
select_method: str = "random"
|
|
"""Method for the selection of factors implementation"""
|
|
|
|
python_bin: str = "python"
|
|
"""Path to the Python binary"""
|
|
|
|
|
|
def get_factor_env(
|
|
conf_type: Optional[str] = None,
|
|
extra_volumes: dict = {},
|
|
running_timeout_period: int = 600,
|
|
enable_cache: Optional[bool] = None,
|
|
) -> Env:
|
|
conf = FactorCoSTEERSettings()
|
|
if hasattr(conf, "python_bin"):
|
|
env = LocalEnv(conf=(CondaConf(conda_env_name=os.environ.get("CONDA_DEFAULT_ENV"))))
|
|
env.conf.extra_volumes = extra_volumes.copy()
|
|
env.conf.running_timeout_period = running_timeout_period
|
|
if enable_cache is not None:
|
|
env.conf.enable_cache = enable_cache
|
|
env.prepare()
|
|
return env
|
|
|
|
|
|
FACTOR_COSTEER_SETTINGS = FactorCoSTEERSettings()
|