mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-02 08:44:44 +00:00
fix ds prefix and store key
This commit is contained in:
parent
4f050be67e
commit
02528986e6
1 changed files with 2 additions and 2 deletions
|
@ -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,
|
||||
)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue