Issue with pytest due to PyTorch torch.load() security update

[prsunoresearch]

Thank you for your hard work maintaining this repository. I've found it very useful for my work.

When I try to run pytest with the test files provided in the tests/directory, I encounter an error related to a security change in recent versions of PyTorch (>2.6). The error indicates that torch.load() cannot be used without specifying the weights_only=False argument to the torch.load() calls.
However, this fix is required in many different locations throughout the test files, and there might be a better way to handle this, which I do not know how.
It would be great if you could address this.