search

sb-ai-lab / LightAutoML

Fast and customizable framework for automatic ML model creation (AutoML)
https://developers.sber.ru/portal/products/lightautoml
Apache License 2.0
1.08k stars 47 forks source link

Fix IndexError in test_tabularautoml_nn.py #108

Closed D1MK4real closed 11 months ago

D1MK4real commented 11 months ago

Fix for IndexError in test_tabularautoml_nn.py: pickle doesn`t save dicts in correct order, so it causes IndexError.

return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
  E       IndexError: index out of range in self

https://github.com/sb-ai-lab/LightAutoML/actions/runs/5740442220/job/15558448712?pr=103