jax/pyrefly.toml at amd-main · ROCm/jax · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
python-version = "3.12"

project-includes = [
    "jax/**/*.py*",
    # "tests/**/*.py",
]

project-excludes = [
    "jax/example_libraries/**/*.py",
    "jax/experimental/**/*.py",
]

ignore-missing-imports = [
    "absl.*",
    "flatbuffers.*",
    "flax.*",
    "google.colab.*",
    "hypothesis.*",
    "IPython.*",
    "jax_cuda12_plugin.*",
    "jax_cuda13_plugin.*",
    "jaxlib.*",
    "jraph.*",
    "kubernetes.*",
    "libtpu.*",
    "matplotlib.*",
    "mpi4py.*",
    "nvidia.*",
    "optax.*",
    "opt_einsum.*",
    "portpicker.*",
    "pygments.*",
    "pytest.*",
    "rich.*",
    "setuptools.*",
    "tensorflow.*",
    "tensorflowjs.*",
    "tensorflow_serving.*",
    "tensorstore.*",
    "web_pdb.*",
    "xprof.*",
    "zstandard.*",
]

permissive-ignores = true
untyped-def-behavior = "check-and-infer-return-any"