fix: adjust good/bad prompts hack to match tests

2026-09-25 05:26:05 -07:00 · 2026-09-22 19:26:02 +05:30
parent 598f5f4bd4
commit 033eb4238c
1 changed files with 2 additions and 2 deletions
@@ -413,13 +413,13 @@ def run():
    # TODO: Introduce a dedicated dataset setting for test prompts.
    good_prompts_dataset = DatasetSpecification(
        dataset="mlabonne/harmless_alpaca",
-        split="train[:400]",
+        split="train[:5]",
        column="text",
    )

    bad_prompts_dataset = DatasetSpecification(
        dataset="mlabonne/harmful_behaviors",
-        split="train[:400]",
+        split="train[:5]",
        column="text",
    )