pyproject.toml

[project]
name = "forgeenv"
version = "0.1.0"
description = "Self-improving RL environment for HuggingFace ecosystem repair under library drift"
requires-python = ">=3.10"
authors = [{name = "akhiilll"}]
license = {text = "Apache-2.0"}
readme = "README.md"
keywords = ["openenv", "reinforcement-learning", "self-play", "code-repair", "huggingface"]

dependencies = [
    "fastapi>=0.110.0",
    "uvicorn>=0.27.0",
    "pydantic>=2.6.0",
    "pyyaml>=6.0",
    "rich>=13.7.0",
    "nltk>=3.8.0",
    "scikit-learn>=1.4.0",
    "numpy>=1.26.0",
]

[project.optional-dependencies]
openenv = [
    "openenv-core>=0.2.0",
]
training = [
    "torch>=2.1.0",
    "transformers>=4.40.0",
    "datasets>=2.18.0",
    "trl>=0.10.0",
    "peft>=0.10.0",
    "accelerate>=0.30.0",
    "wandb>=0.16.0",
]
unsloth = [
    "unsloth>=2024.4",
]
sandbox = [
    "docker>=7.0.0",
]
dev = [
    "pytest>=8.0.0",
    "pytest-asyncio>=0.23.0",
    "matplotlib>=3.8.0",
]

[project.urls]
Homepage = "https://huggingface.co/spaces/akhiilll/forgeenv"
Model = "https://huggingface.co/akhiilll/forgeenv-repair-agent"

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
include = ["forgeenv*"]
exclude = ["tests*", "notebooks*", "artifacts*", "warmstart*", "forgeenv-space*"]

[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py"]
addopts = "-v --tb=short"