fix ferc to eia connection

catalyst-cooperative · Sep 11, 2023 · 836672b · 836672b
1 parent b4901f4
commit 836672b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git 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
@@ -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":