Skip to main content

Browse the archive

Revision f6108674d93727fb89236a51ddc947b0f9d7a17d authored by George Necula on 11 April 2020, 15:27:14 UTC, committed by George Necula on 11 April 2020, 16:30:44 UTC 
Improve error message for indexing into an empty dimension
 
It seems that JAX (and XLA) are a lot more forgiving for index-out-of-bounds,
except for the case when the dimension has size 0. In this case give a similar
IndexError message as numpy, instead of the current one:

   RuntimeError: Invalid argument: Slice size at index 0 in gather op is out
   of range, must be within [0, 1), got 1.:
   This is a bug in JAX's shape-checking rules; please report it!

Issue: #2671
1 parent c4dd1cf
Raw File
mypy.ini 
[mypy]

[mypy-absl.*]
ignore_missing_imports = True
[mypy-jaxlib.*]
ignore_missing_imports = True
[mypy-numpy.*]
ignore_missing_imports = True
[mypy-opt_einsum.*]
ignore_missing_imports = True
[mypy-scipy.*]
ignore_missing_imports = True
[mypy-jax.interpreters.autospmd]
ignore_errors = True
back to top