Readme and loading for chat tasks, float scoring for logviewer render (rllm-org#143)

* Readme and loading for chat tasks, float scoring for logviewer render

* ruff

Author: alexandrasouly-aisi

src/inspect_evals/agentharm/README.md

@@ -11,7 +11,7 @@ Eric Winsor<sup>2</sup>, Jerome Wynne<sup>2</sup>, Yarin Gal<sup>2,♯</sup>, Xa
 
 **Paper**: <https://arxiv.org/abs/2410.09024>
 
-**Benchmark**: <https://huggingface.co/datasets/ai-safety-institute/AgentHarm>
+**Dataset**: <https://huggingface.co/datasets/ai-safety-institute/AgentHarm>
 
 *Note*: At the moment, we release 44 out of 66 public test base behaviors (176 augmented ones) and 8 out of 11 validation base behaviors (32 augmented ones). We hope to release additional behaviors in the future. The average harm score for direct requests (i.e., without jailbreaks) on this initial set of behaviors slightly increases, e.g., from 48.4% to 54.9% on GPT-4o and from 82.2% to 83.6% on Mistral.
 
@@ -29,8 +29,9 @@ First, install the inspect_ai and inspect_evals Python packages with:
 pip install inspect_ai
 pip install git+https://github.com/UKGovernmentBEIS/inspect_evals
 ```
+In VsCode, make sure to download the Inspect AI extension for viewing logs. For more details see https://inspect.ai-safety-institute.org.uk/log-viewer.html.
 
-Examples of running evaluations on two latest proprietary models:
+Examples of running evaluations on two proprietary models:
 ```bash
 inspect eval inspect_evals/agentharm --model openai/gpt-4o-2024-08-06
 inspect eval inspect_evals/agentharm --model anthropic/claude-3-5-sonnet-20240620
@@ -40,6 +41,12 @@ The same but on benign behaviors:
 inspect eval inspect_evals/agentharm_benign --model openai/gpt-4o-2024-08-06
 inspect eval inspect_evals/agentharm_benign --model anthropic/claude-3-5-sonnet-20240620
 ```
+
+Running the chat-only version of the harmful tasks:
+```bash
+inspect eval inspect_evals/agentharm_benign --model openai/gpt-4o-2024-08-06 -T chat_dataset=True
+```
+
 You can also pass arguments to a task to specify using the validation split
 ```bash
 inspect eval inspect_evals/agentharm --model openai/gpt-4o-2024-08-06 -T split=val

src/inspect_evals/agentharm/scorer.py

@@ -113,7 +113,7 @@ async def compute_score(state: TaskState, target: Target) -> Score:
         explanation = f"Grading breakdown: {score_dict}, Refusal explanation: {refusal_explanation}"
 
         combined_score = Score(
-            value={"score": score, "refusal": refusal},
+            value={"score": score, "refusal": 1.0 if refusal else 0.0},
             explanation=explanation,
             metadata={"category": state.metadata["category"]},
         )

src/inspect_evals/agentharm/utils.py

@@ -36,19 +36,19 @@ def load_dataset(
     AGENTHARM_DATASET_LOCATION = (
         Path(user_cache_dir("inspect_evals")) / "agentharm_dataset" / "AgentHarm"
     )
-    snapshot_download(repo_id="ai-safety-institute/AgentHarm", allow_patterns="*.json")
-    if not os.path.exists(AGENTHARM_DATASET_LOCATION):
-        AGENTHARM_DATASET_LOCATION.mkdir(parents=True, exist_ok=True)
-        try:
-            snapshot_download(
-                repo_id="ai-safety-institute/AgentHarm",
-                repo_type="dataset",
-                local_dir=AGENTHARM_DATASET_LOCATION,
-                allow_patterns="*.json",
-            )
-        except Exception as ex:
-            shutil.rmtree(AGENTHARM_DATASET_LOCATION, True)
-            raise ex
+    AGENTHARM_DATASET_LOCATION.mkdir(parents=True, exist_ok=True)
+    try:
+        snapshot_download(
+            repo_id="ai-safety-institute/AgentHarm",
+            repo_type="dataset",
+            local_dir=AGENTHARM_DATASET_LOCATION,
+            allow_patterns="**/*.json",
+            resume_download=True,
+        )
+    except Exception as ex:
+        print(f"Error downloading dataset: {ex}")
+        shutil.rmtree(AGENTHARM_DATASET_LOCATION, ignore_errors=True)
+        raise ex
 
     split_dict = {
         "harmful": {
@@ -73,24 +73,24 @@ def load_dataset(
     if split not in split_dict[name]:
         raise ValueError(f"Split {split} is not available.")
 
-    input_json_file = os.path.join(
-        AGENTHARM_DATASET_LOCATION.as_posix(), split_dict[name][split]
-    )
+    input_json_file = AGENTHARM_DATASET_LOCATION / split_dict[name][split]
 
-    # Path to the output JSONL file
-    output_jsonl_file = input_json_file + "l"
+    # Verify the file exists
+    if not input_json_file.exists():
+        raise FileNotFoundError(f"File not found: {input_json_file}")
 
     # Read the JSON file
     with open(input_json_file, "r") as json_file:
-        data = json.load(json_file)  # This assumes the JSON file contains a JSON array
+        data = json.load(json_file)
 
     # Write to the JSONL file
+    output_jsonl_file = input_json_file.with_suffix(".jsonl")
     with open(output_jsonl_file, "w") as jsonl_file:
         for record in data["behaviors"]:
             jsonl_file.write(json.dumps(record) + "\n")
 
     dataset = json_dataset(
-        output_jsonl_file,
+        str(output_jsonl_file),
         FieldSpec(
             input="prompt",
             id="id",