From 033eb4238c96ffbff2a807e67c04c189f5498c66 Mon Sep 17 00:00:00 2001 From: Philipp Emanuel Weidmann Date: Tue, 22 Sep 2026 19:26:02 +0530 Subject: [PATCH] fix: adjust good/bad prompts hack to match tests --- src/heretic/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/heretic/main.py b/src/heretic/main.py index 3c64fcf..08fb61e 100644 --- a/src/heretic/main.py +++ b/src/heretic/main.py @@ -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", )