ts_forecast - KeyError/ValueError in predict() about flaml

[jccamargo94]

Hi. Is this getting any work? I'm using flaml==2.1.1 and I'm facing the same problem!
I may try to contribute in this issue in some free time at January

[jccamargo94]

Hi. Is this getting any work? I'm using flaml==2.1.1 and I'm facing the same problem! I may try to contribute in this issue in some free time at January

This seems to be a problem with the datetime column as index in the training dataset. I move the index into a column and fit/predict and (de)serialization works as expected.

[EgorKraevTransferwise]

Apologies for taking a while to reply. Can you please also provide code for creating an example dataset of the kind that causes the error (X_* and y_*), so I can reproduce locally?
In general, FLAML TS expects the time to be provided as a column in the input dataframe (time_col argument to fit(), defaulting to the first column of the dataframe). Are you saying if you do this and then the index is also time-valued, the error arises?