fix: update dataset loading in DrawBenchPrompts to retrieve full test set

benchmark/draw_bench.py

@@ -6,7 +6,7 @@ from datasets import load_dataset
 
 class DrawBenchPrompts:
     def __init__(self):
-        self.dataset = load_dataset("shunk031/DrawBench", split="test[:5]")
+        self.dataset = load_dataset("shunk031/DrawBench", split="test")["test"]
 
     def __iter__(self) -> Iterator[Tuple[str, Path]]:
         for i, row in enumerate(self.dataset):