Fix input key expansion in DataLoader.from_parquet (#8248)

okhat · web-flow · commit 694534403b0e · 2025-05-20T15:24:10.000-07:00
diff --git a/dspy/datasets/dataloader.py b/dspy/datasets/dataloader.py
@@ -116,7 +116,7 @@ def from_parquet(
         if not fields:
             fields = list(dataset.features)
 
-        return [dspy.Example({field: row[field] for field in fields}).with_inputs(input_keys) for row in dataset]
+        return [dspy.Example({field: row[field] for field in fields}).with_inputs(*input_keys) for row in dataset]
 
     def from_rm(self, num_samples: int, fields: List[str], input_keys: List[str]) -> List[dspy.Example]:
         try:
