Update dataset in src/lm_buddy/jobs/configs/prometheus.py following S…

…ean's comment Co-authored-by: Sean Friedowitz <[email protected]> Signed-off-by: Davide Eynard <[email protected]>
mozilla-ai · Mar 1, 2024 · de4be8b · de4be8b
1 parent 4e90302
commit de4be8b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/lm_buddy/jobs/configs/prometheus.py b/src/lm_buddy/jobs/configs/prometheus.py
@@ -34,8 +34,7 @@ class PrometheusCompletionsConfig(BaseLMBuddyConfig):
 class PrometheusJobConfig(LMBuddyJobConfig):
     """Configuration to run a prometheus evaluation job."""
 
-    # dataset (json artifact from which we'll extract `text_field`)
-    dataset: TextDatasetConfig
+    dataset: TextDatasetConfig = Field(..., description="dataset (json artifact from which we'll extract `text_field`)")
     # details for our self-hosted prometheus endpoint
     prometheus: PrometheusCompletionsConfig
     # wandb experiment tracking details