Entity tables incorrectly don't load year data

.github/workflows/pr_code_changes.yaml

@@ -50,13 +50,5 @@ jobs:
             run: make test
             env:
               HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
-          - name: Update tests
-            run: make update-tests
-            env:
-              HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
-          - name: Save dataset
-            run: make update-tests
-            env:
-              HUGGING_FACE_TOKEN: ${{ secrets.HUGGING_FACE_TOKEN }}
           - name: Test documentation builds
             run: make documentation

changelog_entry.yaml

@@ -0,0 +1,4 @@
+- bump: patch
+  changes:
+    fixed:
+    - Bug in loading entity tables.

policyengine_uk/data/dataset_schema.py

@@ -48,7 +48,7 @@ def __init__(
                 self.person = f["person"]
                 self.benunit = f["benunit"]
                 self.household = f["household"]
-                self.time_period = f["time_period"].iloc[0]
+                self.time_period = str(f["time_period"].iloc[0])
         else:
             if person is None or benunit is None or household is None:
                 raise ValueError(
@@ -57,9 +57,9 @@ def __init__(
             self.person = person
             self.benunit = benunit
             self.household = household
+            self.time_period = str(fiscal_year)
 
         self.data_format = "arrays"
-        self.time_period = fiscal_year
         self.tables = (self.person, self.benunit, self.household)
 
     def save(self, file_path: str):