from setuptools import setup, find_packages

setup(
    name="red_team_blue_team",
    version="0.1.0",
    description="MBPP red-team-blue-team experiments for password locked shutdown project",
    author="MATS Research",
    packages=find_packages(where="src"),
    package_dir={"": "src"},
    package_data={
        "redteam_strats": ["dataset/*.jsonl"],  # Include JSONL dataset files
    },
    install_requires=[
        "inspect-ai",
        "datasets",
        "openai",
        "docker",
        "pydantic",
        "matplotlib",
        "pandas",
    ],
    python_requires=">=3.11",
)