Skip to content

Commit c899938

Browse files
JashGjgulabrai
authored andcommitted
fix: Set provider_id in NVIDIA notebook when registering dataset (llamastack#3472)
# What does this PR do? When registering a dataset for NVIDIA, the DatasetsRoutingTable expects `nvidia` to be passed via the `provider_id` [here](https://github.com/llamastack/llama-stack/blob/main/llama_stack/core/routing_tables/datasets.py#L61). This PR fixes a notebook to correctly use `provider_id`. <!-- If resolving an issue, uncomment and update the line below --> Closes llamastack#3308 ## Test Plan Manually execute the notebook steps to verify the dataset is registered. Co-authored-by: Jash Gulabrai <[email protected]>
1 parent bda2c63 commit c899938

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/notebooks/nvidia/tool_calling/2_finetuning_and_inference.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -373,7 +373,7 @@
373373
" metadata={\n",
374374
" \"format\": \"json\",\n",
375375
" \"description\": \"Tool calling xLAM dataset in OpenAI ChatCompletions format\",\n",
376-
" \"provider\": \"nvidia\"\n",
376+
" \"provider_id\": \"nvidia\"\n",
377377
" }\n",
378378
")\n",
379379
"print(response)"

0 commit comments

Comments
 (0)