Spaces:
Sleeping
Sleeping
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.
|
| 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] = {
|