From 8c97e2c1c8d623bc5d0bcb5d2f1387393e61ead3 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Mon, 11 Nov 2024 17:13:54 -0800 Subject: [PATCH] fix after rebase --- .../providers/adapters/datasetio/huggingface/huggingface.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/adapters/datasetio/huggingface/huggingface.py b/llama_stack/providers/adapters/datasetio/huggingface/huggingface.py index 7d367a810..cd143a3ef 100644 --- a/llama_stack/providers/adapters/datasetio/huggingface/huggingface.py +++ b/llama_stack/providers/adapters/datasetio/huggingface/huggingface.py @@ -15,7 +15,7 @@ from llama_stack.providers.utils.datasetio.url_utils import get_dataframe_from_u 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): return hf_datasets.load_dataset(**dataset_def.metadata) @@ -41,7 +41,7 @@ class HuggingfaceDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate): async def register_dataset( self, - dataset_def: DatasetDef, + dataset_def: Dataset, ) -> None: self.dataset_infos[dataset_def.identifier] = dataset_def