From c4b9b3cb52c0b6603c59881e20d4641ac448bf82 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 26 Dec 2024 18:11:10 -0800 Subject: [PATCH] huggingface --- .../providers/remote/datasetio/huggingface/huggingface.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py index 2fde7c3d0..47a63677e 100644 --- a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py +++ b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py @@ -5,11 +5,11 @@ # the root directory of this source tree. from typing import Any, Dict, List, Optional -from llama_stack.apis.datasetio import * # noqa: F403 - - import datasets as hf_datasets +from llama_stack.apis.datasetio import DatasetIO, PaginatedRowsResult +from llama_stack.apis.datasets import Dataset + from llama_stack.providers.datatypes import DatasetsProtocolPrivate from llama_stack.providers.utils.datasetio.url_utils import get_dataframe_from_url from llama_stack.providers.utils.kvstore import kvstore_impl