phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 23:29:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

naming fix

Browse source
This commit is contained in:
Xi Yan 2024-10-22 15:57:47 -07:00
parent f131c28db7
commit d6df949049
3 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/distribution/routers/routing_tables.py
View file

@ -217,7 +217,7 @@ class DatasetsRoutingTable(CommonRoutingTableImpl, Datasets):
async def get_dataset( async def get_dataset(
self, dataset_identifier: str self, dataset_identifier: str
) -> Optional[ModelDefWithProvider]: ) -> Optional[DatasetDefWithProvider]:
return self.get_object_by_identifier(identifier) return self.get_object_by_identifier(identifier)
async def register_dataset(self, dataset_def: DatasetDefWithProvider) -> None: async def register_dataset(self, dataset_def: DatasetDefWithProvider) -> None:

4
llama_stack/providers/impls/meta_reference/datasetio/__init__.py
View file

@ -11,8 +11,8 @@ async def get_provider_impl(
config: MetaReferenceDatasetIOConfig, config: MetaReferenceDatasetIOConfig,
_deps, _deps,
): ):
from .datasetio import MetaReferenceDatasetioImpl from .datasetio import MetaReferenceDatasetIOImpl
impl = MetaReferenceDatasetioImpl(config) impl = MetaReferenceDatasetIOImpl(config)
await impl.initialize() await impl.initialize()
return impl return impl

2
llama_stack/providers/impls/meta_reference/datasetio/datasetio.py
View file

@ -90,7 +90,7 @@ class PandasDataframeDataset(BaseDataset):
self.df = df self.df = df
class MetaReferenceDatasetioImpl(DatasetIO, DatasetsProtocolPrivate): class MetaReferenceDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
def __init__(self, config: MetaReferenceDatasetIOConfig) -> None: def __init__(self, config: MetaReferenceDatasetIOConfig) -> None:
self.config = config self.config = config
# local registry for keeping track of datasets within the provider # local registry for keeping track of datasets within the provider