From 32d025923c722dea102490b86061e41ddc4f8a6a Mon Sep 17 00:00:00 2001
From: Yang Yang <yyy@meta.com>
Date: Wed, 12 Mar 2025 12:39:07 -0700
Subject: [PATCH] change to new schema for URL() codegen

---
 llama_stack/templates/open-benchmark/open_benchmark.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/llama_stack/templates/open-benchmark/open_benchmark.py b/llama_stack/templates/open-benchmark/open_benchmark.py
index b78418d55..17f5b8ee7 100644
--- a/llama_stack/templates/open-benchmark/open_benchmark.py
+++ b/llama_stack/templates/open-benchmark/open_benchmark.py
@@ -229,7 +229,7 @@ def get_distribution_template() -> DistributionTemplate:
         DatasetInput(
             dataset_id="bfcl",
             provider_id="huggingface",
-            url={"uri": "https://huggingface.co/datasets/llamastack/bfcl_v3"},
+            url=URL(uri="https://huggingface.co/datasets/llamastack/bfcl_v3"),
             metadata={
                 "path": "llamastack/bfcl_v3",
                 "split": "train",