pyproject.toml

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

[project]
name = "reasonbudget-gym"
version = "0.1.0"
description = "RL environment for learning to allocate reasoning compute under budget constraints"
requires-python = ">=3.10"
dependencies = [
    "fastapi>=0.110.0",
    "uvicorn[standard]>=0.29.0",
    "pydantic>=2.0",
    "numpy>=1.24",
    "datasets>=2.18.0",
    "sentence-transformers>=2.7.0",
    "matplotlib>=3.8",
    "seaborn>=0.13",
]

[project.optional-dependencies]
dev = [
    "pytest>=8.0",
    "httpx>=0.27",
]
train = [
    "torch>=2.2",
]
live = [
    "together>=1.2",
]

[tool.hatch.build.targets.wheel]
packages = ["reasonbudget_gym"]

[tool.pytest.ini_options]
testpaths = ["reasonbudget_gym/tests"]