Support dict for dataclass_with_defaults (used by stateclass)

google · Apr 27, 2024 · e61bf29 · e61bf29
1 parent 37d6a89
commit e61bf29
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/mesop/dataclass_utils/dataclass_utils.py b/mesop/dataclass_utils/dataclass_utils.py
@@ -29,6 +29,8 @@ def dataclass_with_defaults(cls: Type[C]) -> Type[C]:
         setattr(cls, name, field(default=False))
       elif get_origin(type_hint) == list:
         setattr(cls, name, field(default_factory=list))
+      elif get_origin(type_hint) == dict:
+        setattr(cls, name, field(default_factory=dict))
       elif isinstance(type_hint, type):
         setattr(
           cls, name, field(default_factory=dataclass_with_defaults(type_hint))

diff --git a/mesop/dataclass_utils/dataclass_utils_test.py b/mesop/dataclass_utils/dataclass_utils_test.py
@@ -43,6 +43,7 @@ class WithPandasDataFrame:
 @dataclass_with_defaults
 class DataclassNoDefaults:
   foo: int
+  dict: dict[str, str]
 
 
 class UnannotatedClass: