Minor bug fixes (#258)

* Fix minor bug * Fix typing bug
PolicyEngine · Aug 28, 2024 · 6a7afa2 · 6a7afa2
1 parent 62e6be0
commit 6a7afa2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/changelog_entry.yaml b/changelog_entry.yaml
@@ -0,0 +1,4 @@
+- bump: minor
+  changes:
+    fixed:
+    - Bugs in typing and Dataset saving.
diff --git a/policyengine_core/data/dataset.py b/policyengine_core/data/dataset.py
@@ -187,7 +187,7 @@ def save_dataset(self, data, file_path: str = None) -> None:
                             f"Could not save {variable} to {file}. The value is {value}."
                         )
         elif self.data_format == Dataset.FLAT_FILE:
-            values.to_csv(file, index=False)
+            data.to_csv(file, index=False)
 
     def load_dataset(
         self,

diff --git a/policyengine_core/types/data_types/arrays.py b/policyengine_core/types/data_types/arrays.py
@@ -3,7 +3,7 @@
 
 T = TypeVar("T", bool, bytes, float, int, object, str)
 
-ArrayLike = ArrayType[Any] | Sequence[T]
+ArrayLike = Sequence[T]
 
 """:obj:`typing.Generic`: Type of any castable to :class:`numpy.ndarray`.