diff --git a/src/pudl/analysis/ferc1_eia.py b/src/pudl/analysis/ferc1_eia.py index daa1c79754..875209644e 100644 --- a/src/pudl/analysis/ferc1_eia.py +++ b/src/pudl/analysis/ferc1_eia.py @@ -107,7 +107,7 @@ def __init__( start_date: Start date that cooresponds to the tables passed in. end_date: End date that cooresponds to the tables passed in. """ - self.plant_parts_eia = plant_parts_eia + self.plant_parts_eia = plant_parts_eia.set_index("record_id_eia") self.plants_all_ferc1 = plants_all_ferc1 self.fbp_ferc1 = fbp_ferc1 diff --git a/src/pudl/analysis/plant_parts_eia.py b/src/pudl/analysis/plant_parts_eia.py index 8624c7dbc0..b14e4bb864 100644 --- a/src/pudl/analysis/plant_parts_eia.py +++ b/src/pudl/analysis/plant_parts_eia.py @@ -393,7 +393,7 @@ def mega_gens_asset(mcoe: pd.DataFrame, own_eia860: pd.DataFrame) -> pd.DataFram "denorm_plants_eia": AssetIn(key="denorm_plants_eia"), "denorm_utilities_eia": AssetIn(key="denorm_utilities_eia"), }, - # io_manager_key=io_manager_key, + io_manager_key=io_manager_key, compute_kind="Python", ) def plant_parts_eia_asset( @@ -642,10 +642,6 @@ def execute(self, gens_mega, plants_eia860, utils_eia860): pandas.DataFrame: The complete plant parts list """ # aggregate everything by each plant part - gens_mega = gens_mega[ - (gens_mega.report_date < "2021-01-01") - & (gens_mega.report_date >= "2020-01-01") - ] part_dfs = [] for part_name in PLANT_PARTS: if part_name == "plant_match_ferc1":