Relax dependency on lightgbm#46
Conversation
|
Hi @jlopezpena, Upgrading to LightGBM v4 won't be a straightforward process. Here's why: Output Shape Changes: The output shapes of LightGBM models have changed across versions. We'll need to verify compatibility with our current code. Optuna Compatibility: Compatibility testing is required to ensure Optuna works seamlessly with LightGBM v4. Downgrading Context: I'll try to check in the near future if I can work on it |
|
Optuna works with lighgbm >= 4 since last summer: optuna/optuna#4844 Is there any test suite I can run to check if anything breaks? |
|
For reference, I just ran my modelling pipeline on my branch, with LightGBM 4.3.0, and it ran successfully without any errors. I am obviously not using every method in ARFS, but I did use the following:
|
Thanks for reviewing! While I don't currently have a dedicated unit test suite, I'm still working on adapting the existing one to function with the new ARFS API. In the meantime, the provided tutorial notebooks act as an integration test suite. I'll be running these tests with the relaxed version of LightGBM. Expect to see an update on the unit test suite by the end of the month. You're right, fastshap is a nice project but isn't yet ready for production purposes. |
Latest version of
lightgbmis 4.3, butarfsis hard-pinning version 3.3.1. This PR relaxes the dependency to allow for more recent lightgbm versions