search

google-deepmind / distrax

Apache License 2.0
535 stars 32 forks source link

[NumPy] Fix test failure in distrax under NumPy 1.23. #182

Closed copybara-service[bot] closed 2 years ago

copybara-service[bot] commented 2 years ago

[NumPy] Fix test failure in distrax under NumPy 1.23.

It appears that this test indexes out of bounds and under NumPy 1.23 it fails. Change the indexing to be in bounds since that seems to be the intent of the test.