dreamquark-ai / tabnet

PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf
https://dreamquark-ai.github.io/tabnet/
MIT License
2.55k stars 470 forks source link

feat: enable torch 2.0 by relaxing poetry #501

Closed Optimox closed 11 months ago

Optimox commented 11 months ago

What kind of change does this PR introduce? This only changes the requirements specified in poetry, in hope to allow torch 2.0 versions

Does this PR introduce a breaking change?

What needs to be documented once your changes are merged? Nothing much Closing issues

`closes #490