registry refactor

This commit is contained in:
Xi Yan 2024-10-14 16:09:55 -07:00
parent c50686b6fe
commit 95fd53d292
8 changed files with 39 additions and 71 deletions

View file

@ -7,5 +7,4 @@ from llama_stack.apis.datasets import * # noqa: F403
from ..registry import Registry
class DatasetRegistry(Registry[BaseDataset]):
_REGISTRY: Dict[str, BaseDataset] = {}
DatasetRegistry = Registry[BaseDataset]()