heat
heat copied to clipboard
[Bug]: Support special versions of pytorch in setup.py
What happened?
Special versions of pytorch adds a plus sign and some text to the version number like 1.12.1+cpu
or 1.12.1+cu116
. When installing Heat, these packages get replaced (1.12.1
).
https://github.com/helmholtz-analytics/heat/blob/5ff03147ae0ff338a07efaefaf3f288e7090c6cc/setup.py#L36
Version
main (development branch), release
PyTorch version
1.12.1
using less rather than less equal for the upper bound fixes the issue for me