From c683ae1958dd87f0e28ace93a1fa15901384bb96 Mon Sep 17 00:00:00 2001
From: Matteo Prandi <matteo.prandi94@gmail.com>
Date: Wed, 20 May 2026 09:50:25 +0200
Subject: [PATCH] fix: pin default Hugging Face dataset revision

---
 README.md                    |  2 ++
 src/ahb_inspect/constants.py |  1 +
 src/ahb_inspect/dataset.py   |  8 ++++++--
 src/ahb_inspect/tasks.py     |  3 ++-
 tests/test_dataset.py        | 23 +++++++++++++++++++++++
 tests/test_tasks.py          | 23 +++++++++++++++++++++++
 6 files changed, 57 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index fa20957..1a95209 100644
--- a/README.md
+++ b/README.md
@@ -127,6 +127,8 @@ uv run inspect eval ahb_inspect/ahb \
 
 The Inspect task loads `icaro-lab/ahb`, split `test`, maps each row's `prompt` into an Inspect sample, and preserves `method`, `hazard`, `persona`, `locale`, `source_id`, `prompt_hash`, and `canary` as sample metadata. The scorer uses the public AHB safe/unsafe judge rubric and reports `attack_success_rate` as the mean fraction of responses classified `UNSAFE`.
 
+By default, the Inspect task pins the Hugging Face dataset revision to `007d95844bcdd8e8d6a9c4f943214c5981f9186e`. To evaluate a different dataset commit intentionally, pass `-T dataset_revision=<sha>` to `inspect eval`.
+
 ## Citation
 
 If you use AHB in your research, please cite the benchmark. The full citation is available in [`citation.txt`](./citation.txt).
diff --git a/src/ahb_inspect/constants.py b/src/ahb_inspect/constants.py
index b8e9778..f667adb 100644
--- a/src/ahb_inspect/constants.py
+++ b/src/ahb_inspect/constants.py
@@ -1,6 +1,7 @@
 """Shared public AHB evaluation constants."""
 
 DEFAULT_DATASET = "icaro-lab/ahb"
+DEFAULT_DATASET_REVISION = "007d95844bcdd8e8d6a9c4f943214c5981f9186e"
 DEFAULT_SPLIT = "test"
 DEFAULT_JUDGE_MODEL = "openai/gpt-4o-2024-11-20"
 DEFAULT_MAX_JUDGE_TOKENS = 512
diff --git a/src/ahb_inspect/dataset.py b/src/ahb_inspect/dataset.py
index 5e31590..5fde132 100644
--- a/src/ahb_inspect/dataset.py
+++ b/src/ahb_inspect/dataset.py
@@ -6,7 +6,11 @@
 
 from inspect_ai.dataset import Dataset, Sample, hf_dataset
 
-from .constants import DEFAULT_DATASET, DEFAULT_SPLIT
+from ahb_inspect.constants import (
+    DEFAULT_DATASET,
+    DEFAULT_DATASET_REVISION,
+    DEFAULT_SPLIT,
+)
 
 
 METADATA_FIELDS = (
@@ -35,7 +39,7 @@ def record_to_sample(record: dict[str, Any]) -> Sample:
 def load_ahb_dataset(
     dataset: str = DEFAULT_DATASET,
     split: str = DEFAULT_SPLIT,
-    revision: str | None = None,
+    revision: str | None = DEFAULT_DATASET_REVISION,
     limit: int | None = None,
 ) -> Dataset:
     """Load AHB from Hugging Face as Inspect samples."""
diff --git a/src/ahb_inspect/tasks.py b/src/ahb_inspect/tasks.py
index 5a2535d..fc8449c 100644
--- a/src/ahb_inspect/tasks.py
+++ b/src/ahb_inspect/tasks.py
@@ -7,6 +7,7 @@
 
 from ahb_inspect.constants import (
     DEFAULT_DATASET,
+    DEFAULT_DATASET_REVISION,
     DEFAULT_JUDGE_MODEL,
     DEFAULT_SPLIT,
     SYSTEM_PROMPT,
@@ -19,7 +20,7 @@
 def ahb(
     dataset: str = DEFAULT_DATASET,
     split: str = DEFAULT_SPLIT,
-    dataset_revision: str | None = None,
+    dataset_revision: str | None = DEFAULT_DATASET_REVISION,
     dataset_limit: int | None = None,
     use_system_prompt: bool = True,
     judge_model: str = DEFAULT_JUDGE_MODEL,
diff --git a/tests/test_dataset.py b/tests/test_dataset.py
index e8a21bf..d00688b 100644
--- a/tests/test_dataset.py
+++ b/tests/test_dataset.py
@@ -1,3 +1,7 @@
+from inspect_ai.dataset import MemoryDataset, Sample
+
+import ahb_inspect.dataset as dataset_module
+from ahb_inspect.constants import DEFAULT_DATASET_REVISION
 from ahb_inspect.dataset import record_to_sample
 
 
@@ -28,3 +32,22 @@ def test_record_to_sample_maps_prompt_and_metadata() -> None:
         "prompt_hash": "abc123",
         "canary": "ahb:canary",
     }
+
+
+def test_load_ahb_dataset_pins_default_revision(monkeypatch) -> None:
+    calls = {}
+
+    def fake_hf_dataset(**kwargs):
+        calls.update(kwargs)
+        return MemoryDataset([Sample(id="ahb-001", input="prompt")])
+
+    monkeypatch.setattr(dataset_module, "hf_dataset", fake_hf_dataset)
+
+    loaded = dataset_module.load_ahb_dataset(limit=1)
+
+    assert isinstance(loaded, MemoryDataset)
+    assert calls["path"] == "icaro-lab/ahb"
+    assert calls["split"] == "test"
+    assert calls["revision"] == DEFAULT_DATASET_REVISION
+    assert calls["limit"] == 1
+    assert calls["trust"] is False
diff --git a/tests/test_tasks.py b/tests/test_tasks.py
index 56146e4..6d2c199 100644
--- a/tests/test_tasks.py
+++ b/tests/test_tasks.py
@@ -1,6 +1,7 @@
 from inspect_ai.dataset import MemoryDataset, Sample
 
 import ahb_inspect.tasks as tasks
+from ahb_inspect.constants import DEFAULT_DATASET_REVISION
 
 
 def test_ahb_task_uses_public_dataset_parameters(monkeypatch) -> None:
@@ -36,3 +37,25 @@ def fake_load_ahb_dataset(dataset, split, revision, limit):
         "limit": 1,
     }
     assert task.metadata["scoring"] == "attack_success_rate is the mean fraction of responses judged unsafe"
+
+
+def test_ahb_task_pins_default_dataset_revision(monkeypatch) -> None:
+    calls = {}
+
+    def fake_load_ahb_dataset(dataset, split, revision, limit):
+        calls.update(
+            {
+                "dataset": dataset,
+                "split": split,
+                "revision": revision,
+                "limit": limit,
+            }
+        )
+        return MemoryDataset([Sample(id="ahb-001", input="prompt")])
+
+    monkeypatch.setattr(tasks, "load_ahb_dataset", fake_load_ahb_dataset)
+
+    task = tasks.ahb(use_system_prompt=False, judge_model="mockllm/model")
+
+    assert calls["revision"] == DEFAULT_DATASET_REVISION
+    assert task.metadata["dataset_revision"] == DEFAULT_DATASET_REVISION