Merge branch 'usds:main' into main

This commit is contained in:
Saran Ahluwalia 2021-12-05 09:24:42 -05:00 committed by GitHub
commit f1df3d3c34
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.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,
}
)