From de4be8b03ff8f3f66edbea90fbe1cb0184e21732 Mon Sep 17 00:00:00 2001 From: Davide Eynard Date: Fri, 1 Mar 2024 13:21:11 +0000 Subject: [PATCH] Update dataset in src/lm_buddy/jobs/configs/prometheus.py following Sean's comment Co-authored-by: Sean Friedowitz Signed-off-by: Davide Eynard --- src/lm_buddy/jobs/configs/prometheus.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/lm_buddy/jobs/configs/prometheus.py b/src/lm_buddy/jobs/configs/prometheus.py index 5b7f7d05..80fea5cb 100644 --- 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