google / trax

Trax — Deep Learning with Clear Code and Speed
Apache License 2.0
8.01k stars 813 forks source link

[NumPy] Fix uses of deprecated multidimensional NumPy indexing with a non-tuple index. #1755

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

[NumPy] Fix uses of deprecated multidimensional NumPy indexing with a non-tuple index.

NumPy 1.23 removes support for non-tuple indexing of NumPy arrays (https://numpy.org/devdocs/release/1.23.0-notes.html#expired-deprecations). The workaround is to convert multidimensional indices to a tuple.