pyproject.toml

[tool.poetry]
name = "esquilax"
version = "2.0.0"
description = "JAX multi-agent simulation and ML toolset"
authors = [
    "Zombie-Einstein <zombie-einstein@proton.me>"
]
readme = "README.md"
packages = [{include = "esquilax", from = "src"}]
repository = "https://github.com/zombie-einstein/esquilax"
homepage = "https://zombie-einstein.github.io/esquilax"
documentation = "https://zombie-einstein.github.io/esquilax"
license = "MIT"
keywords = [
  "agent-based modelling",
  "simulation",
  "reinforcement-learning",
]
classifiers = [
  "Development Status :: 4 - Beta",
  "Programming Language :: Python",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: Implementation :: CPython",
  "Programming Language :: Python :: Implementation :: PyPy",
  "License :: OSI Approved :: MIT License",
  "Environment :: GPU",
  "Intended Audience :: Science/Research",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
  "Topic :: Scientific/Engineering :: Artificial Life",
]

[tool.poetry.dependencies]
python = "^3.10"
jax = "^0.4.30"
chex = "^0.1.86"
jax-tqdm = "^0.3.1"
evosax = "^0.1.6"
flax = "^0.8.5"

[tool.poetry.group.dev.dependencies]
pre-commit = "^3.7.1"
taskipy = "^1.13.0"
pytest = "^8.2.2"
sphinx = "^8.0.2"
piccolo-theme = "^0.23.0"
sphinx-autoapi = "^3.3.1"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.taskipy.tasks]
test = { cmd = "pytest tests", help="Run tests"}
lint = { cmd = "pre-commit run --all-files", help="Code format checks"}
docs = { cmd = "sphinx-build -W -b html docs/source docs/build", help="Build docs"}
doc-tests = { cmd = "sphinx-build -W -b doctest docs/source docs/build", help="Run sphinx doc-tests"}
pre-commit = "task test && task lint && task docs && task doc-tests"