JAXAtari_Centipede/pyproject.toml at master · zCrazyLion/JAXAtari_Centipede · 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
[project]
name = "jaxatari"
version = "0.1.0"
description = "GPU-accelerated, object-centric Atari environments for reinforcement learning with JAX."
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
    "absl-py>=2.3",
    "ale-py>=0.11.1",
    "chex>=0.1.87",
    "flax",
    "gymnasium>=1.2.0",
    "gymnax>=0.0.8",
    "jax",
    "ml-dtypes",
    "numpy",
    "opt-einsum>=3.4.0",
    "platformdirs>=4.5.1",
    "requests>=2.32.5",
    "scipy>=1.15.3",
    "toolz>=1.0.0",
    "typing-extensions>=4.14.0",
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project.scripts]
install-sprites = "jaxatari.install_sprites:download_and_extract"

[project.optional-dependencies]
dev = [
  "pytest",
  "pygame==2.5.0",
  "gymnasium[other]==1.2.0",
  "syrupy==4.9.1",
]
gh_ci = [
  "pygame==2.5.0",
  "opencv-python-headless",
  "pytest",
  "pytest-sugar",
  "pytest-xdist",
  "pytest-github-actions-annotate-failures",
  "syrupy==4.9.1",
]
training = [
  "tyro",
  "torch",
  "tensorboard",
  "rtpt>=0.0.4",
  "wandb[media]>=0.24.0",
  "hydra-core>=1.3.2",
  "omegaconf>=2.3.0",
  "safetensors>=0.7.0",
]

[tool.pytest.ini_options]
markers = [
  "requires_snapshot: marks tests that rely on snapshot artifacts",
  "smoke: fast checks for PR validation",
  "integration: broader integration checks",
  "slow: expensive checks for exhaustive validation",
  "wrapper_full: tests that should run with the full wrapper recipe matrix",
  "serial: tests that must run without xdist parallel workers",
]