From d6131a1e86e7c7769a71a981f4099ac4c37adb6e Mon Sep 17 00:00:00 2001
From: Nikhil Woodruff <nikhil.woodruff@outlook.com>
Date: Tue, 19 Aug 2025 09:15:00 +0100
Subject: [PATCH] Fix issue in imputation model

---
 changelog_entry.yaml              | 4 ++++
 policyengine_uk_data/utils/qrf.py | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/changelog_entry.yaml b/changelog_entry.yaml
index e69de29bb..4cc707cf0 100644
--- a/changelog_entry.yaml
+++ b/changelog_entry.yaml
@@ -0,0 +1,4 @@
+- bump: patch
+  changes:
+    fixed:
+    - Imputation model syntax.
diff --git a/policyengine_uk_data/utils/qrf.py b/policyengine_uk_data/utils/qrf.py
index 98665889c..d99e5a257 100644
--- a/policyengine_uk_data/utils/qrf.py
+++ b/policyengine_uk_data/utils/qrf.py
@@ -64,7 +64,7 @@ def predict(self, X):
         Returns:
             Predictions at the 0.5 quantile (median).
         """
-        return self.model.predict(X)[0.5]
+        return self.model.predict(X)
 
     def save(self, file_path: str):
         """