From 02528986e6fab2747389785ee53a8bb1aa670d50 Mon Sep 17 00:00:00 2001 From: Vladislav Date: Thu, 5 Dec 2024 17:10:45 +0100 Subject: [PATCH] fix ds prefix and store key --- llama_stack/providers/inline/datasetio/localfs/datasetio.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/inline/datasetio/localfs/datasetio.py b/llama_stack/providers/inline/datasetio/localfs/datasetio.py index f02401088..5b29db126 100644 --- a/llama_stack/providers/inline/datasetio/localfs/datasetio.py +++ b/llama_stack/providers/inline/datasetio/localfs/datasetio.py @@ -22,7 +22,7 @@ from llama_stack.providers.utils.kvstore import kvstore_impl from .config import LocalFSDatasetIOConfig -DATASETS_PREFIX = "locallfs_datasets:" +DATASETS_PREFIX = "datasets:" class BaseDataset(ABC): @@ -119,7 +119,7 @@ class LocalFSDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate): value=dataset.json(), ) dataset_impl = PandasDataframeDataset(dataset) - self.dataset_infos[key] = DatasetInfo( + self.dataset_infos[dataset.identifier] = DatasetInfo( dataset_def=dataset, dataset_impl=dataset_impl, )