Add missing variable types

Author: Mauko Quiroga

openfisca_core/indexed_enums/enum.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import enum
-from typing import Any, Union
+from typing import Any, List, Union
 
 import numpy
 
@@ -227,6 +227,9 @@ def encode(cls, array: A) -> EnumArray:
 
         """
 
+        conditions: List[ArrayType[bool]]
+        choices: List[int]
+
         if isinstance(array, EnumArray):
 
             return array

openfisca_core/indexed_enums/enum_array.py

@@ -76,6 +76,7 @@ def __new__(
             ) -> EnumArray:
         """See comment above."""
 
+        obj: EnumArray
         obj = numpy.asarray(input_array).view(cls)
         obj.possible_values = possible_values
         return obj