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",