Tuple multiply literal

[hatal175]

Description

Fixes #863.

I've added a method hook that given an int literal multiplied by a tuple gives the correct specific sized tuple.

Test Plan

I've added more python evaulation tests and mypy_primer should cover enough of a regression test.

[JelleZijlstra]

mypy-primer found it crashing at https://github.com/google/jax/blob/master/jax/_src/lax/other.py#L96, apparently on a Tuple[int, ...].

[hatal175]

Fixed in commit. I think at least. It doesn't comment if no diff.

…

On Fri, Apr 23, 2021, 17:12 Jelle Zijlstra ***@***.***> wrote: mypy-primer found it crashing at https://github.com/google/jax/blob/master/jax/_src/lax/other.py#L96, apparently on a Tuple[int, ...]. — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub <#10361 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/ACYQXXOEHDGRBDFPVU3MTNDTKF56LANCNFSM43OR2TWQ> .

[JukkaL]

Thanks for the PR! Looks good, I'd just prefer to have these tests use the test fixtures instead of full stubs, since tests using full stubs are a lot slower to run.

[hatal175]

Done as requested.

[JukkaL]

Thanks for the updates! Looks good based on a quick review pass.

Since we are (hopefully) close to a mypy release, I won't merge this until I've cut a release branch. Changes to type inference frequently have hard-to-find edge cases, so I'd prefer to merge this early in a release process to that there will be more time for testing.