mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-17 23:57:15 +00:00
unregister api for dataset
Test Plan: **Type checker and check that the build compiles** **Unit Tests** **E2E Tests** // Screenshots and videos | Before | After | |--| | … | … |
This commit is contained in:
parent
2137b0af40
commit
58d664ab31
9 changed files with 134 additions and 0 deletions
|
|
@ -63,6 +63,11 @@ class HuggingfaceDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
|
|||
)
|
||||
self.dataset_infos[dataset_def.identifier] = dataset_def
|
||||
|
||||
async def unregister_dataset(self, dataset_id: str) -> None:
|
||||
key = f"{DATASETS_PREFIX}{dataset_id}"
|
||||
await self.kvstore.delete(key=key)
|
||||
del self.dataset_infos[dataset_id]
|
||||
|
||||
async def get_rows_paginated(
|
||||
self,
|
||||
dataset_id: str,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue