diff --git a/kis_trader/backtest/optuna_breakout.py b/kis_trader/backtest/optuna_breakout.py index f09f705..094a06e 100644 --- a/kis_trader/backtest/optuna_breakout.py +++ b/kis_trader/backtest/optuna_breakout.py @@ -428,18 +428,25 @@ def run_breakout_optuna( bind_study_trials(study, n_trials=n_trials, log=logger) n_trials = clamp_optimize_n_trials(study, n_trials, log=logger) - import uuid - rust_session_id = f"optuna_break_{uuid.uuid4().hex[:8]}" - try: - import kis_rust_core - import json - candles_json = json.dumps(ctx.codes_candles) - kis_rust_core.init_backtest_session_json( - rust_session_id, - candles_json - ) - except Exception as e: - logger.warning(f"Failed to init_rust_session: {e}") + # 2026-09-06 (docs/rust_engine_parity_port_plan.md · 룰 29): env 게이트로 Rust 세션 초기화 제한. + _use_rust_env = os.environ.get("BACKTEST_USE_RUST") == "1" + rust_session_id: Optional[str] = None + if _use_rust_env: + import uuid + rust_session_id = f"optuna_break_{uuid.uuid4().hex[:8]}" + try: + import kis_rust_core + import json + candles_json = json.dumps(ctx.codes_candles) + kis_rust_core.init_backtest_session_json( + rust_session_id, + candles_json + ) + except Exception as e: + logger.warning(f"Failed to init_rust_session: {e}") + rust_session_id = None + else: + logger.info("🐍 [Python 엔진] Rust 세션 미초기화 (BACKTEST_USE_RUST != 1)") def objective(trial: optuna.Trial) -> float: if ctx.mode == "tpe": @@ -450,8 +457,9 @@ def run_breakout_optuna( ) else: combo = suggest_breakout_params(trial, ctx.mode) - - combo["_rust_session_id"] = rust_session_id + + if rust_session_id is not None: + combo["_rust_session_id"] = rust_session_id result = evaluate_breakout_param_combo( combo, base_fixed=ctx.base_fixed, diff --git a/kis_trader/backtest/optuna_momentum.py b/kis_trader/backtest/optuna_momentum.py index 4623080..269880f 100644 --- a/kis_trader/backtest/optuna_momentum.py +++ b/kis_trader/backtest/optuna_momentum.py @@ -481,18 +481,25 @@ def run_momentum_optuna( bind_study_trials(study, n_trials=n_trials, log=logger) n_trials = clamp_optimize_n_trials(study, n_trials, log=logger) - import uuid - rust_session_id = f"optuna_mome_{uuid.uuid4().hex[:8]}" - try: - import kis_rust_core - import json - candles_json = json.dumps(ctx.codes_candles) - kis_rust_core.init_backtest_session_json( - rust_session_id, - candles_json - ) - except Exception as e: - logger.warning(f"Failed to init_rust_session: {e}") + # 2026-09-06 (docs/rust_engine_parity_port_plan.md · 룰 29): env 게이트로 Rust 세션 초기화 제한. + _use_rust_env = os.environ.get("BACKTEST_USE_RUST") == "1" + rust_session_id: Optional[str] = None + if _use_rust_env: + import uuid + rust_session_id = f"optuna_mome_{uuid.uuid4().hex[:8]}" + try: + import kis_rust_core + import json + candles_json = json.dumps(ctx.codes_candles) + kis_rust_core.init_backtest_session_json( + rust_session_id, + candles_json + ) + except Exception as e: + logger.warning(f"Failed to init_rust_session: {e}") + rust_session_id = None + else: + logger.info("🐍 [Python 엔진] Rust 세션 미초기화 (BACKTEST_USE_RUST != 1)") def objective(trial: optuna.Trial) -> float: if ctx.mode == "tpe": @@ -501,8 +508,9 @@ def run_momentum_optuna( ) else: combo = suggest_momentum_params(trial, ctx.mode, market=getattr(ctx, "market", "KR") or "KR") - - combo["_rust_session_id"] = rust_session_id + + if rust_session_id is not None: + combo["_rust_session_id"] = rust_session_id result = evaluate_momentum_param_combo( combo, base_fixed=ctx.base_fixed, diff --git a/kis_trader/backtest/optuna_scalping.py b/kis_trader/backtest/optuna_scalping.py index 2fa4a5c..2e90b76 100644 --- a/kis_trader/backtest/optuna_scalping.py +++ b/kis_trader/backtest/optuna_scalping.py @@ -415,26 +415,35 @@ def run_scalp_optuna( bind_study_trials(study, n_trials=n_trials, log=logger) n_trials = clamp_optimize_n_trials(study, n_trials, log=logger) - import uuid - rust_session_id = f"optuna_scalp_{uuid.uuid4().hex[:8]}" - try: - import kis_rust_core - import json - candles_json = json.dumps(ctx.codes_candles) - kis_rust_core.init_backtest_session_json( - rust_session_id, - candles_json - ) - except Exception as e: - logger.warning(f"Failed to init_rust_session: {e}") + # 2026-09-06 (docs/rust_engine_parity_port_plan.md · 룰 29): Rust 세션은 BACKTEST_USE_RUST=1 + # 일 때만 초기화하고 combo 에 세팅. 이전엔 env 무관하게 항상 세팅 → Python 선택해도 Rust 로 돌던 버그. + _use_rust_env = os.environ.get("BACKTEST_USE_RUST") == "1" + rust_session_id: Optional[str] = None + if _use_rust_env: + import uuid + rust_session_id = f"optuna_scalp_{uuid.uuid4().hex[:8]}" + try: + import kis_rust_core + import json + candles_json = json.dumps(ctx.codes_candles) + kis_rust_core.init_backtest_session_json( + rust_session_id, + candles_json + ) + except Exception as e: + logger.warning(f"Failed to init_rust_session: {e}") + rust_session_id = None + else: + logger.info("🐍 [Python 엔진] Rust 세션 미초기화 (BACKTEST_USE_RUST != 1)") def objective(trial: optuna.Trial) -> float: if ctx.mode == "tpe": combo = suggest_scalp_params_tpe(trial) else: combo = suggest_scalp_params(trial, ctx.mode) - - combo["_rust_session_id"] = rust_session_id + + if rust_session_id is not None: + combo["_rust_session_id"] = rust_session_id result = evaluate_scalp_param_combo( combo, base_fixed=ctx.base_fixed, diff --git a/kis_trader/backtest/optuna_web_jobs.py b/kis_trader/backtest/optuna_web_jobs.py index ebf73be..d4dfb35 100644 --- a/kis_trader/backtest/optuna_web_jobs.py +++ b/kis_trader/backtest/optuna_web_jobs.py @@ -3756,8 +3756,10 @@ def start_optuna_job( # 사후 report 게이트 거래수 = 탐색과 동일(기간 자동) env["PARAM_SEARCH_OPTUNA_REPORT_MIN_TRADES"] = str(min_trades) env["SORT_BY"] = sort_by - if use_rust: - env["BACKTEST_USE_RUST"] = "0" + # 2026-09-06 정정: 이전 로직 `if use_rust: env["BACKTEST_USE_RUST"] = "0"` 은 완전히 반대. + # use_rust=True → "1" (Rust 활성) / False → "0" (Python 강제). 항상 명시 세팅해 + # 부모 프로세스 env 상속 오염을 차단한다 (룰 29 · docs/rust_engine_parity_port_plan.md). + env["BACKTEST_USE_RUST"] = "1" if use_rust else "0" _labels = { "momentum": "모멘텀", diff --git a/kis_trader/backtest/param_search_optuna.py b/kis_trader/backtest/param_search_optuna.py index d778858..69ce9d4 100644 --- a/kis_trader/backtest/param_search_optuna.py +++ b/kis_trader/backtest/param_search_optuna.py @@ -510,22 +510,30 @@ def run_tail_optuna( bind_study_trials(study, n_trials=n_trials, log=logger) n_trials = clamp_optimize_n_trials(study, n_trials, log=logger) - import uuid - rust_session_id = f"optuna_tail_{uuid.uuid4().hex[:8]}" - try: - from kis_trader.engine.tail_engine import init_rust_session, clear_rust_session - init_rust_session(rust_session_id, ctx.candles_by_code, ctx.ticks_by_code) - except Exception as e: - logger.warning(f"Failed to init_rust_session: {e}") + # 2026-09-06 (docs/rust_engine_parity_port_plan.md · 룰 29): env 게이트로 Rust 세션 초기화 제한. + _use_rust_env = os.environ.get("BACKTEST_USE_RUST") == "1" + rust_session_id: Optional[str] = None + if _use_rust_env: + import uuid + rust_session_id = f"optuna_tail_{uuid.uuid4().hex[:8]}" + try: + from kis_trader.engine.tail_engine import init_rust_session, clear_rust_session # noqa: F401 + init_rust_session(rust_session_id, ctx.candles_by_code, ctx.ticks_by_code) + except Exception as e: + logger.warning(f"Failed to init_rust_session: {e}") + rust_session_id = None + else: + logger.info("🐍 [Python 엔진] Rust 세션 미초기화 (BACKTEST_USE_RUST != 1)") def objective(trial: optuna.Trial) -> float: if ctx.mode == "tpe": combo = suggest_tail_params_tpe(trial, entry_mode=ctx.tpe_entry_mode) else: combo = suggest_tail_params(trial, ctx.mode) - - # Rust 세션 ID를 파라미터에 넘겨서 하위에서 사용할 수 있게 함 - combo["_rust_session_id"] = rust_session_id + + # Rust 세션이 없으면 세팅하지 않음 (Python 경로) + if rust_session_id is not None: + combo["_rust_session_id"] = rust_session_id result = evaluate_tail_param_combo( combo, base_params=ctx.base_params, @@ -748,11 +756,13 @@ def run_tail_optuna( finally: # mode_combo 실측이 ticks 공유뷰를 쓰므로 optimize 직후 unlink 금지 release_shared_tick_store(ctx, log=logger) - try: - from kis_trader.engine.tail_engine import clear_rust_session - clear_rust_session(rust_session_id) - except Exception as e: - pass + # rust_session_id 는 use_rust=True 로 초기화된 경우에만 clear 필요 + if rust_session_id is not None: + try: + from kis_trader.engine.tail_engine import clear_rust_session + clear_rust_session(rust_session_id) + except Exception: + pass def apply_best_trial(study: optuna.Study, ctx: TailSearchContext) -> bool: