Fix Enum bugs

PolicyEngine · Jul 31, 2024 · 02b5695 · 02b5695
1 parent 4b98c95
commit 02b5695
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/policyengine_core/enums/enum.py b/policyengine_core/enums/enum.py
@@ -53,7 +53,7 @@ def encode(cls, array: Union[EnumArray, np.ndarray]) -> EnumArray:
         # Confusingly, Numpy uses "S" to refer to byte-string arrays
         # and "U" to refer to Unicode-string arrays, which are also
         # referred to as the "str" type
-        if array.dtype.kind == "S":
+        if array.dtype.kind == "S" or array.dtype == object:
             # Convert boolean array to string array
             array = array.astype(str)
 

diff --git a/policyengine_core/simulations/simulation.py b/policyengine_core/simulations/simulation.py
@@ -362,6 +362,7 @@ def build_from_dataset(self) -> None:
                 else:
                     entity_level_data = data[variable]
 
+                print(f"Setting {variable} in {time_period} to {data[variable]}")
                 self.set_input(variable, time_period, entity_level_data)
 
         self.default_calculation_period = (