Hi- I would make a PR for this, but I wanted to check that this is something you would accept. There are quite a few improvements from the pinned version to 3.0.0. The version 2.3.1 is over a year old. Is there a reason it hasn't been updated? This is an extremely popular library on Kaggle.