Made PyTorch version check compatible with post-release versions

MouseLand / cellpose

a generalist algorithm for cellular segmentation with human-in-the-loop capabilities

https://www.cellpose.org/

BSD 3-Clause "New" or "Revised" License

1.4k stars 403 forks source link

Open MeyerBender opened 1 week ago

MeyerBender commented 1 week ago

Replaced

major_version, minor_version, _ = torch.__version__.split(".")

with

torch_version = torch.__version__.split(".")
major_version, minor_version = torch_version[0], torch_version[1]

in dynamics.py. Addresses issue #1054, where cellpose would fail when a post-release version of PyTorch was installed.