From 2936133f95b5b5bb90e34e27630643434c53a7da Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Mon, 25 Nov 2024 18:55:54 -0800 Subject: [PATCH] precommit --- llama_stack/providers/remote/datasetio/huggingface/config.py | 3 ++- .../providers/remote/datasetio/huggingface/huggingface.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/datasetio/huggingface/config.py b/llama_stack/providers/remote/datasetio/huggingface/config.py index 46470ce49..1cdae0625 100644 --- a/llama_stack/providers/remote/datasetio/huggingface/config.py +++ b/llama_stack/providers/remote/datasetio/huggingface/config.py @@ -3,12 +3,13 @@ # # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. +from pydantic import BaseModel + from llama_stack.distribution.utils.config_dirs import RUNTIME_BASE_DIR from llama_stack.providers.utils.kvstore.config import ( KVStoreConfig, SqliteKVStoreConfig, ) -from pydantic import BaseModel class HuggingfaceDatasetIOConfig(BaseModel): diff --git a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py index 8d34df672..c2e4506bf 100644 --- a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py +++ b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py @@ -9,6 +9,7 @@ from llama_stack.apis.datasetio import * # noqa: F403 import datasets as hf_datasets + 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