Evgueni Poloukarov commited on
Commit
d2b88b2
·
1 Parent(s): dbf2a77

fix: use renamed 'target' column instead of 'target_border_X' in inference

Browse files

The dynamic_forecast.prepare_forecast_data() method renames the target column
from 'target_border_X' to just 'target', but the inference code was still trying
to access the original column name, causing KeyError.

This was the root cause of empty forecasts - the model never received target data.

src/forecasting/chronos_inference.py CHANGED
@@ -169,8 +169,8 @@ class ChronosInferencePipeline:
169
  border=border
170
  )
171
 
172
- # Get target column name
173
- target_col = f"target_border_{border}"
174
 
175
  # Extract context values
176
  context = context_data[target_col].values
 
169
  border=border
170
  )
171
 
172
+ # Get target column name (note: dynamic_forecast renames it to 'target')
173
+ target_col = 'target'
174
 
175
  # Extract context values
176
  context = context_data[target_col].values