pyproject.toml

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

[project]
name = "redteamml"
version = "0.1.0"
description = "Toolkit for adversarial robustness evaluation of large language models."
readme = "README.md"
license = {text = "MIT"}
requires-python = ">=3.9"
dependencies = [
    "numpy>=1.23",
    "scipy>=1.10",
    "pandas>=1.5",
    "matplotlib>=3.6",
    "tqdm>=4.65",
    "rich>=13.7",
    "typer>=0.9",
    "pydantic>=1.10",
    "scikit-learn>=1.2",
    "networkx>=3.0",
    "sentence-transformers>=2.2",
    "transformers>=4.35",
    "requests>=2.28",
    "tenacity>=8.2",
    "pyyaml>=6.0",
    "polars>=0.20",
    "pyarrow>=15.0",
]

[project.optional-dependencies]
dev = ["pytest>=7.4", "pytest-asyncio>=0.21", "black>=23.9", "ruff>=0.3"]
plotly = ["plotly>=5.20"]
ray = ["ray[default]>=2.5"]

[project.scripts]
rtml = "redteamml.cli.main:main"

[tool.setuptools.packages.find]
include = ["redteamml", "redteamml.*"]