from setuptools import setup, find_packages

setup(
    name="sensitive-attribute-circuit-analysis",
    version="0.1.0",
    description="Mechanistic interpretability study: Do LLMs route through sensitive-attribute circuits that shift analytical choices?",
    author="ryanyen22",
    packages=find_packages(),
    python_requires=">=3.9",
    install_requires=[
        "torch>=2.0.0",
        "transformers>=4.40.0",
        "accelerate>=0.27.0",
        "bitsandbytes>=0.43.0",
        "scikit-learn>=1.3.0",
        "numpy>=1.24.0",
        "matplotlib>=3.7.0",
        "huggingface_hub>=0.20.0",
    ],
    extras_require={
        "nnsight": ["nnsight>=0.3.0"],
        "pyvene": ["pyvene>=0.1.0"],
        "viz": ["circuitsvis>=1.0.0", "plotly>=5.0.0"],
        "all": ["nnsight>=0.3.0", "pyvene>=0.1.0", "circuitsvis>=1.0.0"],
    },
)