def rename_columns(df: pd.DataFrame) -> pd.DataFrame:
    """
    Rename columns to match our schema.
    """

    data = df.copy()

    # Drop irrelevant columns.
    data.drop(columns=["HourDK"], inplace=True)

    # Rename columns
    data.rename(
        columns={
            "HourUTC": "datetime_utc",
            "PriceArea": "area",
            "ConsumerType_DE35": "consumer_type",
            "TotalCon": "energy_consumption",
        },
        inplace=True,
    )

    return data

    Rename columns to match our schema.
    
feature-pipeline/feature_pipeline/etl/cleaning.py