DoE fix to address #975 (#979)

* Fixed input field name

Co-authored-by: Saran Ahluwalia <sarahluw@cisco.com>
This commit is contained in:
Saran Ahluwalia 2021-12-05 08:26:39 -05:00 committed by GitHub
commit 610343a1e3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -20,12 +20,13 @@ class DOEEnergyBurden(ExtractTransformLoad):
) )
self.TRACT_INPUT_COLUMN_NAME = "FIP" 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 # Constants for output
self.COLUMNS_TO_KEEP = [ self.COLUMNS_TO_KEEP = [
self.GEOID_TRACT_FIELD_NAME, self.GEOID_TRACT_FIELD_NAME,
self.ENERGY_BURDEN_FIELD_NAME, self.REVISED_ENERGY_BURDEN_FIELD_NAME
] ]
self.raw_df: pd.DataFrame self.raw_df: pd.DataFrame
@ -56,7 +57,7 @@ class DOEEnergyBurden(ExtractTransformLoad):
output_df = self.raw_df.rename( output_df = self.raw_df.rename(
columns={ 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, self.TRACT_INPUT_COLUMN_NAME: self.GEOID_TRACT_FIELD_NAME,
} }
) )