[build-system]
requires = ["setuptools>=45", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "openenv-brain-rl-region-selection"
version = "0.1.0"
description = "OpenEnv environment for RL-based sequential brain region selection on auditory fMRI summaries"
requires-python = ">=3.10"
dependencies = [
    "openenv-core[core] @ git+https://github.com/meta-pytorch/OpenEnv.git@v0.2.3",
    "fastapi>=0.115.0",
    "pydantic>=2.0.0",
    "uvicorn>=0.24.0",
    "requests>=2.31.0",
    "numpy>=1.26",
    "pandas>=2.0",
]

[project.optional-dependencies]
dev = [
    "pytest>=8.0.0",
]
atlas = [
    "nilearn>=0.10",
    "nibabel>=5.0",
]
plots = [
    "matplotlib>=3.7",
]
deploy = [
    "huggingface_hub>=0.24",
]
policy = [
    "huggingface_hub>=0.24",
    "transformers>=4.46,<4.54",
    "accelerate",
    "peft",
]
llm-rl = [
    "trl>=0.19,<0.20",
    "datasets",
    "transformers>=4.46,<4.54",
    "accelerate",
    "peft",
]

[project.scripts]
brainrl-server = "brain_rl_region_selection.server.app:main"

[tool.setuptools]
include-package-data = true
packages = ["brain_rl_region_selection", "brain_rl_region_selection.server"]
package-dir = { "brain_rl_region_selection" = ".", "brain_rl_region_selection.server" = "server" }