From 5bc5d224c282fc1764e28abdb273d210c9204fb6 Mon Sep 17 00:00:00 2001 From: Insider77Circle <160362522+Insider77Circle@users.noreply.github.com> Date: Sat, 7 Feb 2026 15:04:50 -0500 Subject: [PATCH] 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> --- .../cross_asset_correlation/correlation_regimes_fixed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tradingagents/cross_asset_correlation/correlation_regimes_fixed.py b/tradingagents/cross_asset_correlation/correlation_regimes_fixed.py index 07445a15..0c86be2b 100644 --- a/tradingagents/cross_asset_correlation/correlation_regimes_fixed.py +++ b/tradingagents/cross_asset_correlation/correlation_regimes_fixed.py @@ -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: