Fix Extended FRS bug (#21)

CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.2.3] - 2024-09-18 12:57:28
+
+### Fixed
+
+- Bug causing the Extended FRS to not generate.
+
 ## [1.2.2] - 2024-09-18 11:38:21
 
 ## [1.2.1] - 2024-09-18 10:05:45
@@ -33,6 +39,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 
 
+[1.2.3]: https://github.com/PolicyEngine/policyengine-us-data/compare/1.2.2...1.2.3
 [1.2.2]: https://github.com/PolicyEngine/policyengine-us-data/compare/1.2.1...1.2.2
 [1.2.1]: https://github.com/PolicyEngine/policyengine-us-data/compare/1.2.0...1.2.1
 [1.2.0]: https://github.com/PolicyEngine/policyengine-us-data/compare/1.1.0...1.2.0

changelog.yaml

@@ -23,3 +23,8 @@
     removed:
     - Survey Enhance is no longer used as a top-level import.
   date: 2024-09-18 11:38:21
+- bump: patch
+  changes:
+    fixed:
+    - Bug causing the Extended FRS to not generate.
+  date: 2024-09-18 12:57:28

policyengine_uk_data/datasets/frs/extended_frs.py

@@ -68,6 +68,8 @@ def generate(self):
             ["age", "gender", "region"]
         )
         create_income_model()
+        from survey_enhance import Imputation
+
         income = Imputation.load(STORAGE_FOLDER / "income.pkl")
         full_imputations = income.predict(income_inputs)
         for variable in full_imputations.columns:

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "policyengine_uk_data"
-version = "1.2.2"
+version = "1.2.3"
 description = "A package to create representative microdata for the UK."
 readme = "README.md"
 authors = [