update the device for is_valid_x

[bkmi]

This PR should address issues in the tests mentioned in #788. The is_valid_x index will be (maybe) copied to the correct device for x and theta before indexing. Thereby a runtime error is avoided.

[codecov-commenter]

Codecov Report

Merging #789 (5c0bf36) into main (8fe45c9) will not change coverage.
The diff coverage is 100.00%.

@@           Coverage Diff           @@
##             main     #789   +/-   ##
=======================================
  Coverage   74.52%   74.52%           
=======================================
  Files          77       77           
  Lines        6120     6120           
=======================================
  Hits         4561     4561           
  Misses       1559     1559           

Flag	Coverage Δ
unittests	74.52% <100.00%> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
sbi/inference/snle/snle_base.py	91.58% <100.00%> (ø)
sbi/inference/snpe/snpe_base.py	89.17% <100.00%> (ø)
sbi/inference/snre/snre_base.py	94.87% <100.00%> (ø)

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

[bkmi]

this commit fixes the tests that were failing but makes pyright fail

[janfb]

great, thanks for fixing this!

The pyright issues seem unrelated, I will look into it. Once tests are passing , this is good to go in.