diff --git a/llama_stack/providers/inline/datasetio/localfs/__init__.py b/llama_stack/providers/inline/datasetio/localfs/__init__.py index afcbb85e8..db8aa555c 100644 --- a/llama_stack/providers/inline/datasetio/localfs/__init__.py +++ b/llama_stack/providers/inline/datasetio/localfs/__init__.py @@ -4,15 +4,15 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. -from .config import LocalfsDatasetIOConfig +from .config import LocalFSDatasetIOConfig async def get_provider_impl( - config: LocalfsDatasetIOConfig, + config: LocalFSDatasetIOConfig, _deps, ): - from .datasetio import LocalfsDatasetIOImpl + from .datasetio import LocalFSDatasetIOImpl - impl = LocalfsDatasetIOImpl(config) + impl = LocalFSDatasetIOImpl(config) await impl.initialize() return impl diff --git a/llama_stack/providers/inline/datasetio/localfs/config.py b/llama_stack/providers/inline/datasetio/localfs/config.py index dfdce5056..58d563c99 100644 --- a/llama_stack/providers/inline/datasetio/localfs/config.py +++ b/llama_stack/providers/inline/datasetio/localfs/config.py @@ -6,4 +6,4 @@ from llama_stack.apis.datasetio import * # noqa: F401, F403 -class LocalfsDatasetIOConfig(BaseModel): ... +class LocalFSDatasetIOConfig(BaseModel): ... diff --git a/llama_stack/providers/inline/datasetio/localfs/datasetio.py b/llama_stack/providers/inline/datasetio/localfs/datasetio.py index b797e1e16..d8c100684 100644 --- a/llama_stack/providers/inline/datasetio/localfs/datasetio.py +++ b/llama_stack/providers/inline/datasetio/localfs/datasetio.py @@ -15,7 +15,7 @@ from dataclasses import dataclass from llama_stack.providers.datatypes import DatasetsProtocolPrivate from llama_stack.providers.utils.datasetio.url_utils import get_dataframe_from_url -from .config import LocalfsDatasetIOConfig +from .config import LocalFSDatasetIOConfig class BaseDataset(ABC): @@ -77,8 +77,8 @@ class PandasDataframeDataset(BaseDataset): self.df = self._validate_dataset_schema(df) -class LocalfsDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate): - def __init__(self, config: LocalfsDatasetIOConfig) -> None: +class LocalFSDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate): + def __init__(self, config: LocalFSDatasetIOConfig) -> None: self.config = config # local registry for keeping track of datasets within the provider self.dataset_infos = {} diff --git a/llama_stack/providers/registry/datasetio.py b/llama_stack/providers/registry/datasetio.py index 7b05120ad..2d1c722f0 100644 --- a/llama_stack/providers/registry/datasetio.py +++ b/llama_stack/providers/registry/datasetio.py @@ -16,7 +16,7 @@ def available_providers() -> List[ProviderSpec]: provider_type="localfs", pip_packages=["pandas"], module="llama_stack.providers.inline.datasetio.localfs", - config_class="llama_stack.providers.inline.datasetio.localfs.LocalfsDatasetIOConfig", + config_class="llama_stack.providers.inline.datasetio.localfs.LocalFSDatasetIOConfig", api_dependencies=[], ), remote_provider_spec(