automl.fit_predict() raises KeyError: 'index' when used with latest version of pandas==2.1.1

[refaat31]

Explanation

I am getting a KeyError when executing the automl.fit_predict() function while following this notebook - https://colab.research.google.com/github/AILab-MLTools/LightAutoML/blob/master/examples/tutorials/Tutorial_1_basics.ipynb

error trace

version used:

Python 3.10.12 lightautoml==0.3.8b1 pandas==2.1.1

References:

• Reference notebook showing the output of the error : https://colab.research.google.com/drive/1X-hLB7O4wF8gyU6gBd09KzYq83GTIF1E

• With pandas==1.4.3, however, I can get the same code to run without any errors raised. - https://colab.research.google.com/drive/1S-WRjWMCfM99kS5ZMvlVJoLpwcWQiTL2

[fingoldo]

Очень отзывчивые разработчики...

[alexmryzhkov]

Hi @fingoldo,

Currently we are working on migration to pandas 2.x but yes, this process takes longer than all of us expected because of the dramatic changes in pandas lib.

Alex