d5d30240fe71a09c45b45a719230b6c6f83c9404,pudl/analysis.py,,simple_select,#Any#Any#,18

Before Change

    // Creates a sql Select object
    select = sa.sql.select([tbl, ])
    // Converts sql object to pandas dataframe
    return(pd.read_sql(select, pudl_engine))


def simple_ferc1_plant_ids(pudl_engine):
    

After Change

    if "plant_id" in table.columns:
        // Shorthand for readability... pt = PUDL Tables
        pt = models.PUDLBase.metadata.tables

        // Pull in plants_eia which connects EIA & PUDL plant IDs
        plants_eia_tbl = pt["plants_eia"]
        plants_eia_select = sa.sql.select([
            plants_eia_tbl.c.plant_id,
            plants_eia_tbl.c.plant_id_pudl,
        ])
        plants_eia = pd.read_sql(plants_eia_select, pudl_engine)
        out_df = pd.merge(table, plants_eia, how="left", on="plant_id")
        out_df.rename(columns={"plant_id": "plant_id_eia"}, inplace=True)
        out_df.plant_id_pudl = out_df.plant_id_pudl.astype(int)
        table = out_df
    else:
        table = table

Instances