Update starknet_py/utils/typed_data.py

Co-authored-by: ddoktorski <45050160+ddoktorski@users.noreply.github.com>
software-mansion · Jul 5, 2024 · 0085160 · 0085160
1 parent 7f5798f
commit 0085160
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/starknet_py/utils/typed_data.py b/starknet_py/utils/typed_data.py
@@ -263,9 +263,7 @@ def _verify_types(self):
         referenced_types = set()
         for type_name in self.types:
             for ref_type in self.types[type_name]:
-                if isinstance(ref_type, MerkleTreeParameter):
-                    referenced_types.add(ref_type.contains)
-                elif isinstance(ref_type, EnumParameter):
+                if isinstance(ref_type, (EnumParameter, MerkleTreeParameter)):
                     referenced_types.add(ref_type.contains)
                 elif is_enum_variant_type(ref_type.type):
                     referenced_types.update(_extract_enum_types(ref_type.type))