From 1381d3f3e8ed2709c5ce64be6055335dc78ddc43 Mon Sep 17 00:00:00 2001 From: raspawar Date: Wed, 9 Apr 2025 12:34:36 +0000 Subject: [PATCH] linting fix --- .../remote/datasetio/nvidia/README.md | 23 +++++++++---------- 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/llama_stack/providers/remote/datasetio/nvidia/README.md b/llama_stack/providers/remote/datasetio/nvidia/README.md index 699894ca5..1d3d15132 100644 --- a/llama_stack/providers/remote/datasetio/nvidia/README.md +++ b/llama_stack/providers/remote/datasetio/nvidia/README.md @@ -29,12 +29,14 @@ llama stack build --template nvidia --image-type conda ```python import os + os.environ["NVIDIA_API_KEY"] = "your-api-key" os.environ["NVIDIA_CUSTOMIZER_URL"] = "http://nemo.test" os.environ["NVIDIA_USER_ID"] = "llama-stack-user" os.environ["NVIDIA_DATASET_NAMESPACE"] = "default" os.environ["NVIDIA_PROJECT_ID"] = "test-project" from llama_stack.distribution.library_client import LlamaStackAsLibraryClient + client = LlamaStackAsLibraryClient("nvidia") client.initialize() ``` @@ -43,17 +45,14 @@ client.initialize() ```python client.datasets.register( -purpose="post-training/messages", -dataset_id="my-training-dataset", -source={ -"type": "uri", -"uri": "hf://datasets/default/sample-dataset" -}, -metadata={ -"format": "json", -"description": "Dataset for LLM fine-tuning", -"provider": "nvidia" -} + purpose="post-training/messages", + dataset_id="my-training-dataset", + source={"type": "uri", "uri": "hf://datasets/default/sample-dataset"}, + metadata={ + "format": "json", + "description": "Dataset for LLM fine-tuning", + "provider": "nvidia", + }, ) ``` @@ -72,4 +71,4 @@ for dataset in datasets: ```python client.datasets.unregister(dataset_id="my-training-dataset") -``` \ No newline at end of file +```