I get a mypy error when I index into an array using both Ellipsis and tuple unpacking. I have a line of code that looks like this: new_mat = mat[..., *np.ix_(inds, inds)] which gives rise to the following mypy error error: Argument 1 to <tuple> has incompatible type "EllipsisType"; expected "ndarray[tuple[int, ...], dtype[integer[Any] | numpy.bool[builtins.bool]]]" [arg-type] I have similar lines of code several times in close proximity, with different matrices and sets of indices. I would prefer not to sprinkle mypy ignore messages everywhere, or otherwise increase the verbosity of my code. The code is also called with jax arrays (in which case np is really jax.numpy), including by jitted functions. Is there an idiomatic way to resolve the mypy error? Continue reading...
