lerax/pyproject.toml at main · RunnersNum40/lerax · 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
[project]
name = "lerax"
version = "0.0.6"
description = "Deep Reinforcement Learning with JAX and Equinox."
readme = "README.md"
requires-python = ">=3.13"
license = "Apache-2.0"
license-files = ["LICENSE"]
authors = [{ name = "Theodore Pinkerton", email = "ted@tedpinkerton.ca" }]
dependencies = [
  "asciichartpy>=1.5.25",
  "diffrax>=0.7.0",
  "distreqx>=0.0.2",
  "equinox>=0.13.3",
  "flax>=0.12.2",
  "gymnasium>=1.2.3",
  "gymnax>=0.0.9",
  "imageio>=2.37.2",
  "imageio-ffmpeg>=0.6.0",
  "jax>=0.9.0",
  "jaxtyping>=0.3.6",
  "moviepy>=2.2.1",
  "mujoco-mjx>=3.4.0",
  "once-py>=0.0.8",
  "optax>=0.2.6",
  "pygame>=2.6.1",
  "rich>=14.3.1",
  "tensorboard>=2.20.0",
  "tensorboardX>=2.6.4",
  "wandb>=0.25.0",
]

[project.urls]
documentation = "https://lerax.tedpinkerton.ca/"
homepage = "https://lerax.tedpinkerton.ca/"
issues = "https://github.com/RunnersNum40/lerax/issues"
source = "https://github.com/RunnersNum40/lerax.git"

[project.optional-dependencies]
onnx = ["jax2onnx>=0.12.0", "onnx>=1.17.0"]

[dependency-groups]
dev = [
  "gymnasium>=1.2.3",
  "gymnax>=0.0.9",
  "jax2onnx>=0.12.0",
  "mkdocstrings-python>=2.0.1",
  "mujoco-mjx>=3.4.0",
  "onnx>=1.17.0",
  "onnxruntime>=1.19.0",
  "prek>=0.3.0",
  "pygame>=2.6.1",
  "pytest>=9.0.2",
  "pytest-cov>=7.0.0",
  "pytest-xdist>=3.8.0",
  "ruff>=0.14.14",
  "setuptools<71",
  "stable-baselines3>=2.7.1",
  "ty>=0.0.13",
  "zensical>=0.0.19",
]

# Kept for contributors that use pyright rather than ty
[tool.pyright]
reportIncompatibleVariableOverride = false  # Incompatible with eqx.AbstractVar
stubPath = "stubs"
include = ["src/**", "examples/**", "tests/**"]

[tool.pytest]
addopts = ["--cov=lerax", "--import-mode=importlib"]
strict = true
testpaths = ["tests"]

[tool.ruff]
exclude = ["stubs"]

[tool.ruff.format]
docstring-code-format = true

[tool.ruff.lint]
fixable = ["I001", "F401"]
ignore = ["E402", "E721", "E731", "E741", "F722", "E501"]
select = ["E", "F", "I001"]

[tool.setuptools.package-data]
lerax = ["env/mujoco/assets/**", "env/unitree/g1/assets/**"]

[tool.ty]
[tool.ty.rules]

[tool.ty.src]
include = ["src", "examples", "tests"]

[tool.ty.environment]
extra-paths = ["stubs"]

[tool.typos]
[tool.typos.files]
extend-exclude = ["stubs"]

[tool.uv]
package = true

[tool.uv.sources]
equinox = { git = "https://github.com/patrick-kidger/equinox.git", branch = "main" }

[[tool.uv.index]]
name = "testpypi"
url = "https://test.pypi.org/simple/"
publish-url = "https://test.pypi.org/legacy/"
explicit = true