From 610343a1e31e7f76351564686ba243bd42919b64 Mon Sep 17 00:00:00 2001 From: Saran Ahluwalia Date: Sun, 5 Dec 2021 08:26:39 -0500 Subject: [PATCH] DoE fix to address #975 (#979) * Fixed input field name Co-authored-by: Saran Ahluwalia --- .../data_pipeline/etl/sources/doe_energy_burden/etl.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/data/data-pipeline/data_pipeline/etl/sources/doe_energy_burden/etl.py b/data/data-pipeline/data_pipeline/etl/sources/doe_energy_burden/etl.py index 7889d2f6..7bd2878c 100644 --- a/data/data-pipeline/data_pipeline/etl/sources/doe_energy_burden/etl.py +++ b/data/data-pipeline/data_pipeline/etl/sources/doe_energy_burden/etl.py @@ -20,12 +20,13 @@ class DOEEnergyBurden(ExtractTransformLoad): ) self.TRACT_INPUT_COLUMN_NAME = "FIP" - self.ENERGY_BURDEN_FIELD_NAME = "BURDEN" + self.INPUT_ENERGY_BURDEN_FIELD_NAME = "BURDEN" + self.REVISED_ENERGY_BURDEN_FIELD_NAME = "Energy burden" # Constants for output self.COLUMNS_TO_KEEP = [ self.GEOID_TRACT_FIELD_NAME, - self.ENERGY_BURDEN_FIELD_NAME, + self.REVISED_ENERGY_BURDEN_FIELD_NAME ] self.raw_df: pd.DataFrame @@ -56,7 +57,7 @@ class DOEEnergyBurden(ExtractTransformLoad): output_df = self.raw_df.rename( columns={ - "BURDEN": self.ENERGY_BURDEN_FIELD_NAME, + self.INPUT_ENERGY_BURDEN_FIELD_NAME : self.REVISED_ENERGY_BURDEN_FIELD_NAME, self.TRACT_INPUT_COLUMN_NAME: self.GEOID_TRACT_FIELD_NAME, } )