mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 16:07:46 +00:00
TPTBusiness
20428f7d91
- quant.py: guard against empty orch_factors, move strategy_name before try block - quant_proposal.py: fix __init__ return type Tuple[dict,bool] -> None - strategy_orchestrator.py: remove dead rdagent_logger import shadowed by getLogger - factor.py: replace unusual 'not x is None' with idiomatic 'x is not None' - workflow/loop.py: withdraw_loop(0) raises RuntimeError instead of looking for folder -1 - workflow/tracking.py: replace crash-prone AssertionError with logger.warning + skip - factor_from_report.py: fix misleading comment about loop_n/step_n dual use
66 lines
1.9 KiB
Python
Executable File
66 lines
1.9 KiB
Python
Executable File
"""
|
|
Factor workflow with session control
|
|
"""
|
|
|
|
import asyncio
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
import fire
|
|
from rdagent.app.qlib_rd_loop.conf import FACTOR_PROP_SETTING
|
|
from rdagent.components.workflow.rd_loop import RDLoop
|
|
from rdagent.core.exception import CoderError, FactorEmptyError
|
|
from rdagent.log import rdagent_logger as logger
|
|
|
|
|
|
class FactorRDLoop(RDLoop):
|
|
skip_loop_error = (FactorEmptyError, CoderError)
|
|
skip_loop_error_stepname = "feedback"
|
|
|
|
def running(self, prev_out: dict[str, Any]):
|
|
exp = self.runner.develop(prev_out["coding"])
|
|
if exp is None:
|
|
logger.error("Factor extraction failed.")
|
|
raise FactorEmptyError("Factor extraction failed.")
|
|
logger.log_object(exp, tag="runner result")
|
|
return exp
|
|
|
|
|
|
def main(
|
|
path: str | None = None,
|
|
step_n: int | None = None,
|
|
loop_n: int | None = None,
|
|
all_duration: str | None = None,
|
|
checkout: bool = True,
|
|
checkout_path: str | None = None,
|
|
base_features_path: str | None = None,
|
|
**kwargs,
|
|
):
|
|
"""
|
|
Auto R&D Evolving loop for fintech factors.
|
|
|
|
You can continue running session by
|
|
|
|
.. code-block:: python
|
|
|
|
dotenv run -- python rdagent/app/qlib_rd_loop/factor.py $LOG_PATH/__session__/1/0_propose --step_n 1 # `step_n` is a optional paramter
|
|
|
|
"""
|
|
if checkout_path is not None:
|
|
checkout = Path(checkout_path)
|
|
|
|
if path is None:
|
|
factor_loop = FactorRDLoop(FACTOR_PROP_SETTING)
|
|
else:
|
|
factor_loop = FactorRDLoop.load(path, checkout=checkout)
|
|
|
|
factor_loop._init_base_features(base_features_path)
|
|
if "user_interaction_queues" in kwargs and kwargs["user_interaction_queues"] is not None:
|
|
factor_loop._set_interactor(*kwargs["user_interaction_queues"])
|
|
factor_loop._interact_init_params()
|
|
asyncio.run(factor_loop.run(step_n=step_n, loop_n=loop_n, all_duration=all_duration))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
fire.Fire(main)
|