fix type errors

capsulbrasil · Sep 10, 2024 · ea08b15 · ea08b15
1 parent 720520f
commit ea08b15
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/normalize_json/normalize.py b/src/normalize_json/normalize.py
@@ -265,12 +265,14 @@ def translate(
 
             if node.get('enum') and value != None:
                 if node.get('array'):
+                    if not isinstance(value, list):
+                        raise TypeError()
                     value = [
-                        node['enum'].get(v)
-                        for v in value
+                        node.get('enum', {}).get(v)
+                        for v in typing.cast(list[str], value)
                     ]
                 else:
-                    value = node['enum'].get(value)
+                    value = node.get('enum', {}).get(str(typing.cast(typing.Any, value)))
 
             ret[original_name] = value