[project]
name = "general-bots-training"
version = "0.1.0"
requires-python = ">=3.13"
dependencies = [
    "equinox>=0.13.8",
    "generals-bots",
    "jax==0.11.0",
    "jaxtyping>=0.3.11",
    "omegaconf>=2.3.1",
    "optax>=0.2.8",
]

[project.optional-dependencies]
cuda = [
    "jax[cuda]==0.11.0"
]
cuda13 = [
    "jax[cuda13]==0.11.0"
]

[tool.uv.sources]
generals-bots = { git = "https://github.com/strakam/generals-bots.git" }


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

[tool.hatch.build.targets.wheel]
packages = ["src/general_bots_training"]

[tool.ty]
python-path = ["src"]

[tool.black]
line-length = 100

[tool.isort]
line_length = 100
profile = "black"
known_typing = "typing" # types,typing_extensions,mypy,mypy_extensions
sections = "FUTURE,TYPING,STDLIB,THIRDPARTY,FIRSTPARTY,LOCALFOLDER"
# skip_glob = [""]  # files/folders/... to skip
# known_first_party = [""]  # packages that are forced as first party
# src_paths = [""]  # files inside these paths are treated as first party
# multi_line_output = 5
float_to_top = true
group_by_package = true
combine_as_imports = true

[tool.ruff]
line-length = 100

[tool.ruff.lint]
select = ["E", "F"]

[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["F401"]

[dependency-groups]
dev = [
    "ipython>=9.16.1",
    "pytest>=9.1.1",
]