From 7891cada1a04dfe034137097b7112bbe72577739 Mon Sep 17 00:00:00 2001
From: mikivee <mikivee>
Date: Thu, 9 Jan 2025 20:01:31 +0000
Subject: [PATCH] add appliiance cols for whole home

---
 src/datagen.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/datagen.py b/src/datagen.py
index 8e37ff0..3379d36 100644
--- a/src/datagen.py
+++ b/src/datagen.py
@@ -131,6 +131,8 @@ class DataGenerator(tf.keras.utils.Sequence):
         "neighbor_distance_ft",
         "n_occupants",
         "vintage",
+        "has_heat_pump_dryer",
+        "has_induction_range",
         # fuel indicators -- these must be present for post-processing to work!!
         "has_methane_gas_appliance",
         "has_fuel_oil_appliance",