sbi-dev/sbi

Analysis tests file on GPU

Closed this issue · 1 comments

Describe the bug
The test_analysis_modules unit test fails on GPU.

It is due to the tensors being on different devices. (maybe the device is not specified correctly?)

To Reproduce
Run pytest on a GPU.

I am not able to reduce it. I tried

pytest tests/analysis_test.py::test_analysis_modules