Evgueni Poloukarov commited on
Commit
2ea0150
·
1 Parent(s): bf6dbbb

debug: add forecast shape logging to diagnose dimension mismatch

Browse files

- Inference now succeeds but output shape is wrong
- Adding debug logging to see actual vs expected shape
- Bump version to v1.0.6

src/forecasting/chronos_inference.py CHANGED
@@ -2,7 +2,7 @@
2
  """
3
  Chronos-2 Inference Pipeline
4
  Standalone inference script for HuggingFace Space deployment.
5
- FORCE REBUILD: v1.0.5
6
  """
7
 
8
  import os
@@ -186,6 +186,7 @@ class ChronosInferencePipeline:
186
 
187
  # Calculate quantiles
188
  forecast_numpy = forecast.numpy()
 
189
 
190
  # Store results
191
  results['borders'][border] = {
 
2
  """
3
  Chronos-2 Inference Pipeline
4
  Standalone inference script for HuggingFace Space deployment.
5
+ FORCE REBUILD: v1.0.6
6
  """
7
 
8
  import os
 
186
 
187
  # Calculate quantiles
188
  forecast_numpy = forecast.numpy()
189
+ print(f"[DEBUG] Forecast shape: {forecast_numpy.shape}, Expected: ({num_samples}, {prediction_hours})", flush=True)
190
 
191
  # Store results
192
  results['borders'][border] = {