SkinTokens / src /data /spec.py
pookiefoof's picture
Public release: SkinTokens 路 TokenRig demo
9d7cf7f
from abc import ABC, abstractmethod
from dataclasses import fields
class ConfigSpec(ABC):
@classmethod
def check_keys(cls, config, expect=None):
if expect is None:
expect = [field.name for field in fields(cls)] # type: ignore
for key in config.keys():
if key not in expect:
raise ValueError(f"expect names {expect} in {cls.__name__}, found {key}")
@classmethod
@abstractmethod
def parse(cls, **kwargs) -> 'ConfigSpec':
raise NotImplementedError()