bonsai/pyproject.toml at main · jax-ml/bonsai · 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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
[project]
name = "jax-bonsai"
dynamic = ["version"]
description = "A collection of lightweight, open-source models built on JAX."
readme = "README.md"
requires-python = ">=3.11"
license = { text = "Apache-2.0" }
authors = [
  { name = "JAX-bonsai developers", email = "jax-bonsai@google.com" },
]
classifiers = [
  "Development Status :: 3 - Alpha",
  "Intended Audience :: Developers",
  "Intended Audience :: Science/Research",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3.13",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

dependencies = [
  "jax >= 0.8.0, < 0.9.0",
  "jaxlib >= 0.8.0, < 0.9.0",
  # Set flax from source as long as bonsai has no stable release
  # "flax >= 0.12.0, < 0.13.0",
  "flax @ git+https://github.com/google/flax.git",
  "jaxtyping>=0.2.20",
  "jinja2>=3.0.0",
  "huggingface-hub>=0.20.0",
  "transformers >= 4.30.0, < 5.0.0",
]

[project.optional-dependencies]
tf = ["h5py", "keras_hub", "tensorflow"]
vision = ["timm", "pillow>=11.3.0", "opencv-python-headless"]
audio = ["librosa"]

dev = [
  "xprof",
  "pre-commit",
  "jupytext == 1.18.1",
  "ruff",
]

test-env = [
  "torch",
  "timm",
  "h5py",
  "diffusers[flax]",
  "keras_hub",
]

testing = [
  "pytest",
  "pytest-xdist",
  "jax-bonsai[test-env]",
]

[project.urls]
Source = "https://github.com/jax-ml/bonsai"

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
include = ["bonsai*"]

[tool.setuptools.dynamic]
version = {attr = "bonsai.__version__"}

[tool.ruff]
target-version = "py311"
line-length = 120
extend-include = ["*.ipynb"]

[tool.ruff.lint]
ignore = [
    # Unnecessary collection call
    "C408",
    # Unnecessary map usage
    "C417",
    # Unnecessary dict comprehension for iterable
    "C420",
    # Object names too complex
    "C901",
    # Local variable is assigned to but never used
    "F841",
    # Class could be dataclass or namedtuple
    "B903",
    # Raise with from clause inside except block
    "B904",
    # Zip without explicit strict parameter
    "B905",
]
select = [
    "B9",
    "C",
    "F",
    "W",
    "YTT",
    "ASYNC",
    "E101",
]

[tool.ruff.format]
quote-style = "double"
indent-style = "space"

[tool.jupytext]
formats = "ipynb,md:myst"
cell_metadata_filter = "-all"