phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +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(
self, dataset_identifier: str
) -> Optional[ModelDefWithProvider]:
) -> Optional[DatasetDefWithProvider]:
return self.get_object_by_identifier(identifier)
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,
_deps,
):
from .datasetio import MetaReferenceDatasetioImpl
from .datasetio import MetaReferenceDatasetIOImpl
impl = MetaReferenceDatasetioImpl(config)
impl = MetaReferenceDatasetIOImpl(config)
await impl.initialize()
return impl

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

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