From 64fb1905a1ad4d4e5c620577ffdfd095e9b30a78 Mon Sep 17 00:00:00 2001
From: Botao Chen <markchen1015@meta.com>
Date: Tue, 11 Mar 2025 21:02:22 -0700
Subject: [PATCH] refine

---
 llama_stack/templates/open-benchmark/open_benchmark.py | 2 +-
 llama_stack/templates/open-benchmark/run.yaml          | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/llama_stack/templates/open-benchmark/open_benchmark.py b/llama_stack/templates/open-benchmark/open_benchmark.py
index 7f002254b..7df33a715 100644
--- a/llama_stack/templates/open-benchmark/open_benchmark.py
+++ b/llama_stack/templates/open-benchmark/open_benchmark.py
@@ -196,7 +196,7 @@ def get_distribution_template() -> DistributionTemplate:
             url={"uri": "https://huggingface.co/datasets/llamastack/gpqa_0shot_cot"},
             metadata={
                 "path": "llamastack/gpqa_0shot_cot",
-                "name": "main",
+                "name": "gpqa_main",
                 "split": "train",
             },
             dataset_schema={
diff --git a/llama_stack/templates/open-benchmark/run.yaml b/llama_stack/templates/open-benchmark/run.yaml
index b607d57e6..97c54e621 100644
--- a/llama_stack/templates/open-benchmark/run.yaml
+++ b/llama_stack/templates/open-benchmark/run.yaml
@@ -185,7 +185,7 @@ datasets:
     uri: https://huggingface.co/datasets/llamastack/gpqa_0shot_cot
   metadata:
     path: llamastack/gpqa_0shot_cot
-    name: main
+    name: gpqa_main
     split: train
   dataset_id: gpqa_cot
   provider_id: huggingface