Add functionality to set a device fixture depending on the availability

aai-institute / pyDVL

pyDVL is a library of stable implementations of algorithms for data valuation and influence function computation

https://pydvl.org

GNU Lesser General Public License v3.0

89 stars 9 forks source link

Closed schroedk closed 1 month ago

schroedk commented 1 month ago

This PR closes #573

Add a device fixture, which depending on the availability and user input (--with-cuda) resolves to cuda device.

[x] Wrote Unit tests (if necessary)
[x] Updated Documentation (if necessary)
[x] Updated Changelog
[x] ~If notebooks were added/changed, added boilerplate cells are tagged with "tags": ["hide"] or "tags": ["hide-input"]~

schroedk commented 1 month ago

Useful! But wouldn't one want to run tests both with and without cuda as well?

Yes, but I would ask the user to do it manually, and not automatically. In our automatic test environment, we do not have access to cuda anyway.