Handle annotations and enums better in Config

hezarai · Aug 23, 2023 · dc8ee58 · dc8ee58
1 parent a7e7e2c
commit dc8ee58
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/hezar/configs.py b/hezar/configs.py
@@ -70,8 +70,10 @@ def __post_init__(self):
         elif "name" not in self.__annotations__ and self.name is not None:
             self.name: str = self.name
 
-        if self.config_type is not None and isinstance(self.config_type, Enum):
-            self.config_type: str = self.config_type.value
+        # Convert enums to values
+        for param, value in self.dict().items():
+            if isinstance(getattr(self, param), Enum):
+                setattr(self, param, getattr(self, param).value)
 
     def __getitem__(self, item):
         try: