Before submitting a pull request (PR), please read the contributing guide.
Please fill out as much of this template as you can, but if you have any problems or questions, just leave a comment and we will help out :)
Description
What is this PR
[ ] Bug fix
[x] Addition of a new feature
[ ] Other
Why is this PR needed?
Tests in brainglobe-workflows fail on macos-latest as they attempt to run using mps which is not available on github runners even though torch.backends.mps.is_available() returns True.
What does this PR do?
Moves the logic for forcing PyTorch to use the cpu as the device into the cellfinder package. This will hopefully allow the fix from this repo to generalise to brainglobe-workflows.
How has this PR been tested?
Tested on CI (only place I can reproduce this issue).
Is this a breaking change?
No
Does this PR require an update to the documentation?
No
Checklist:
[ ] The code has been tested locally
[ ] Tests have been added to cover all new functionality (unit & integration)
[ ] The documentation has been updated to reflect any changes
Before submitting a pull request (PR), please read the contributing guide.
Please fill out as much of this template as you can, but if you have any problems or questions, just leave a comment and we will help out :)
Description
What is this PR
Why is this PR needed? Tests in
brainglobe-workflowsfail on macos-latest as they attempt to run usingmpswhich is not available on github runners even thoughtorch.backends.mps.is_available()returnsTrue.What does this PR do? Moves the logic for forcing PyTorch to use the cpu as the device into the
cellfinderpackage. This will hopefully allow the fix from this repo to generalise tobrainglobe-workflows.How has this PR been tested?
Tested on CI (only place I can reproduce this issue).
Is this a breaking change?
No
Does this PR require an update to the documentation?
No
Checklist: