BaseModelConfigs now also serializes class variables and not just instance variables to ensure proper restoration.

Author: seidnerj

Tutorials/05_sound_to_text/configs.py

@@ -3,6 +3,7 @@
 
 from mltu.configs import BaseModelConfigs
 
+
 class ModelConfigs(BaseModelConfigs):
     def __init__(self):
         super().__init__()

mltu/configs.py

@@ -1,15 +1,25 @@
 import os
 import yaml
 
+
 class BaseModelConfigs:
     def __init__(self):
         self.model_path = None
 
     def serialize(self):
-        # get object attributes
-        return self.__dict__
+        class_attributes = {key: value
+                            for (key, value)
+                            in type(self).__dict__.items()
+                            if key not in ['__module__', '__init__', '__doc__', '__annotations__']}
+        instance_attributes = self.__dict__
+
+        # first init with class attributes then apply instance attributes, overwriting any existing duplicate attributes
+        all_attributes = class_attributes.copy()
+        all_attributes.update(instance_attributes)
+
+        return all_attributes
 
-    def save(self, name: str="configs.yaml"):
+    def save(self, name: str = "configs.yaml"):
         if self.model_path is None:
             raise Exception("Model path is not specified")