pull/307/head
Kye 1 year ago
parent 8af7ff19bc
commit bcb69e0960

@ -1,4 +0,0 @@
rules:
line-length:
level: warning
allow-non-breakable-inline-mappings: true

@ -1,28 +0,0 @@
from typing import Dict
from pydantic import BaseModel
class Registry(BaseModel):
"""Registry for storing and building classes."""
name: str
entries: Dict = {}
def register(self, key: str):
def decorator(class_builder):
self.entries[key] = class_builder
return class_builder
return decorator
def build(self, type: str, **kwargs):
if type not in self.entries:
raise ValueError(
f"{type} is not registered. Please register with the"
f' .register("{type}") method provided in {self.name} registry'
)
return self.entries[type](**kwargs)
def get_all_entries(self):
return self.entries
Loading…
Cancel
Save