phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-02 08:44:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

linting fix

Browse source
This commit is contained in:
raspawar 2025-04-02 14:26:48 +00:00 • committed by Rashmi Pawar
parent 2baf252f71
commit cf3f3ff130
3 changed files with 5 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/distribution/routers/routing_tables.py
View file

@ -440,7 +440,7 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
# infer provider from source
if metadata:
if metadata.get("provider"):
provider_id = metadata.get("provider") # pass through from nvidia datasetio
provider_id = metadata.get("provider") # pass through from nvidia datasetio
elif source.type == DatasetType.rows.value:
provider_id = "localfs"
elif source.type == DatasetType.uri.value:

6
llama_stack/providers/remote/datasetio/nvidia/datasetio.py
View file

@ -9,12 +9,14 @@ from typing import Any, Dict, List, Optional
import aiohttp
from llama_stack.apis.common.content_types import URL
from llama_stack.apis.common.type_system import ParamType
from llama_stack.apis.common.responses import PaginatedResponse
from llama_stack.apis.common.type_system import ParamType
from llama_stack.apis.datasets import Dataset
from llama_stack.schema_utils import webmethod
from llama_stack.apis.datasets import DatasetPurpose, DataSource, Dataset
from .config import NvidiaDatasetIOConfig
class NvidiaDatasetIOAdapter:
"""Nvidia NeMo DatasetIO API."""

5
llama_stack/templates/nvidia/nvidia.py
View file

@ -52,11 +52,6 @@ def get_distribution_template() -> DistributionTemplate:
model_id="${env.SAFETY_MODEL}",
provider_id="nvidia",
)
datasetio_provider = Provider(
provider_id="nvidia",
provider_type="remote::nvidia",
config=NvidiaDatasetIOConfig.sample_run_config(),
)
available_models = {
"nvidia": MODEL_ENTRIES,