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

fix after rebase

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-11 17:13:54 -08:00
parent 600f23703d
commit 8c97e2c1c8
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/adapters/datasetio/huggingface/huggingface.py
View file

@ -15,7 +15,7 @@ from llama_stack.providers.utils.datasetio.url_utils import get_dataframe_from_u
from .config import HuggingfaceDatasetIOConfig from .config import HuggingfaceDatasetIOConfig
def load_hf_dataset(dataset_def: DatasetDef): def load_hf_dataset(dataset_def: Dataset):
if dataset_def.metadata.get("path", None): if dataset_def.metadata.get("path", None):
return hf_datasets.load_dataset(**dataset_def.metadata) return hf_datasets.load_dataset(**dataset_def.metadata)
@ -41,7 +41,7 @@ class HuggingfaceDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
async def register_dataset( async def register_dataset(
self, self,
dataset_def: DatasetDef, dataset_def: Dataset,
) -> None: ) -> None:
self.dataset_infos[dataset_def.identifier] = dataset_def self.dataset_infos[dataset_def.identifier] = dataset_def