Revision a002643a4aff1803c578b3520464367e7e9e3bac authored by Peter Hawkins on 02 March 2023, 16:23:50 UTC, committed by jax authors on 02 March 2023, 16:24:30 UTC
Work around pytype bug. It seems that the line from functools import cached_property causes pytype to give up on the entire module. Avoid the member import to fix the type inference. PiperOrigin-RevId: 513544106
1 parent a9421a8
| File | Mode | Size |
|---|---|---|
| _src | ||
| example_libraries | ||
| experimental | ||
| image | ||
| interpreters | ||
| lax | ||
| lib | ||
| nn | ||
| numpy | ||
| ops | ||
| scipy | ||
| tools | ||
| BUILD | -rw-r--r-- | 6.4 KB |
| __init__.py | -rw-r--r-- | 7.4 KB |
| abstract_arrays.py | -rw-r--r-- | 763 bytes |
| ad_checkpoint.py | -rw-r--r-- | 711 bytes |
| api_util.py | -rw-r--r-- | 769 bytes |
| cloud_tpu_init.py | -rw-r--r-- | 633 bytes |
| collect_profile.py | -rw-r--r-- | 4.6 KB |
| config.py | -rw-r--r-- | 696 bytes |
| core.py | -rw-r--r-- | 7.0 KB |
| custom_batching.py | -rw-r--r-- | 657 bytes |
| custom_derivatives.py | -rw-r--r-- | 1.2 KB |
| custom_transpose.py | -rw-r--r-- | 644 bytes |
| debug.py | -rw-r--r-- | 935 bytes |
| distributed.py | -rw-r--r-- | 638 bytes |
| dlpack.py | -rw-r--r-- | 653 bytes |
| dtypes.py | -rw-r--r-- | 1.1 KB |
| errors.py | -rw-r--r-- | 1.1 KB |
| flatten_util.py | -rw-r--r-- | 629 bytes |
| jaxpr_util.py | -rw-r--r-- | 6.4 KB |
| linear_util.py | -rw-r--r-- | 1.0 KB |
| monitoring.py | -rw-r--r-- | 1.1 KB |
| prng.py | -rw-r--r-- | 997 bytes |
| profiler.py | -rw-r--r-- | 1.1 KB |
| py.typed | -rw-r--r-- | 0 bytes |
| random.py | -rw-r--r-- | 6.9 KB |
| sharding.py | -rw-r--r-- | 1.4 KB |
| stages.py | -rw-r--r-- | 1.2 KB |
| test_util.py | -rw-r--r-- | 835 bytes |
| tree_util.py | -rw-r--r-- | 2.6 KB |
| typing.py | -rw-r--r-- | 2.8 KB |
| util.py | -rw-r--r-- | 1.1 KB |
| version.py | -rw-r--r-- | 1003 bytes |
Computing file changes ...
