test_all_models indexer bug

sberbank-ai-lab / RePlay

RecSys Library

https://sberbank-ai-lab.github.io/RePlay/

Apache License 2.0

68 stars 6 forks source link

test_all_models indexer bug #87

Closed Darel13712 closed 2 years ago

Darel13712 commented 2 years ago

test_predict_pairs_warm_only multvae fails with IndexError: index 4 is out of bounds for dimension 0 with size 4

AleXXL1986 commented 2 years ago

I find problem: some wrong id are in log df. in function _predict_pairs_wrap you are not filter log df. you need do it correct lines before 629 line in base_rec.py