Update tradingagents/cross_asset_correlation/correlation_regimes_fixed.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Insider77Circle 2026-02-07 15:04:50 -05:00 committed by GitHub
parent d7cee37e01
commit 5bc5d224c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions

View File

@ -203,11 +203,11 @@ class CorrelationRegimeDetector:
Returns:
Tuple of (predicted regime, probability)
"""
if current_regime.value not in transition_matrix.index:
if current_regime not in transition_matrix.index:
return current_regime, 1.0
# Get transition probabilities from current regime
probs = transition_matrix.loc[current_regime.value]
probs = transition_matrix.loc[current_regime]
# Adjust based on market conditions if provided
if market_conditions: