Traceback (most recent call last): File "/home/tanpv/Progs/claude/HW_CFS_raw/src/rt_forecast_ai.py", line 269, in main() File "/home/tanpv/Progs/claude/HW_CFS_raw/src/rt_forecast_ai.py", line 126, in main T = train_ai(w, a.ncore) ^^^^^^^^^^^^^^^^^^^^ File "/home/tanpv/Progs/claude/HW_CFS_raw/src/rt_forecast_ai.py", line 81, in train_ai Ftr = feats(d0[train], doy[train]) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/tanpv/Progs/claude/HW_CFS_raw/src/rt_forecast_ai.py", line 79, in feats return np.stack([d0_row, fc, cos, sin, pi], -1).reshape(-1, 5) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/tanpv/.conda/envs/ehd/lib/python3.11/site-packages/numpy/_core/shape_base.py", line 458, in stack raise ValueError('all input arrays must have the same shape') ValueError: all input arrays must have the same shape