# ============================================================================
# Python package versions of the environment in which the reported runs were
# executed (Python 3.10.0, CUDA 12.1 wheels, NVIDIA A100 80GB).
#
# Pins are taken from the frozen environment of the experiment machine that
# produced the black-box runs and all reported completions/ASR values. Where a
# different version was recorded for an earlier stage of a run, ENVIRONMENT.md
# lists it -- read that file before treating this list as complete.
#
# Install into a fresh Python 3.10 environment:
#     python -m venv .venv && . .venv/bin/activate
#     pip install -r requirements.txt
#     python -m spacy download en_core_web_sm
# Then clone HarmBench (commit in ENVIRONMENT.md) and install its data.
# ============================================================================

# --- core inference / training stack ---
torch==2.4.0
torchvision==0.19.0
transformers==4.50.0
tokenizers==0.21.1
vllm==0.6.3.post1
ray==2.44.0
accelerate==1.5.2
xformers==0.0.27.post2
safetensors==0.5.3
huggingface-hub==0.29.3
sentencepiece==0.2.0
einops==0.8.1

# --- HarmBench harness dependencies ---
# fschat provides the `fastchat` module used for chat templates. The harness
# recorded fastchat 0.2.36 at run time; the pin below is TBD (author to confirm
# which distribution was installed).
# fschat==0.2.36
spacy==3.8.4
spacy-legacy==3.0.12
spacy-loggers==1.0.5
confection==0.1.5
nltk==3.9.1
datasketch==1.6.5          # hash-based classifier for copyright behaviors
Jinja2==3.1.0
numpy==1.26.4
scipy==1.15.2
pandas==2.2.3
protobuf==5.29.4

# --- API clients (only needed for the GPT-4 judge/target of TAP-Transfer) ---
openai==1.68.2
anthropic==0.49.0
mistralai==1.6.0
google-generativeai==0.8.4
