diff --git a/etl/route_march/oo_prs_additional_units/oo_prs_additional_units.py b/etl/route_march/oo_prs_additional_units/oo_prs_additional_units.py index c1b562ea..69e08f9a 100644 --- a/etl/route_march/oo_prs_additional_units/oo_prs_additional_units.py +++ b/etl/route_march/oo_prs_additional_units/oo_prs_additional_units.py @@ -131,10 +131,14 @@ def rename_and_add_columns(epcs): } ) + epcs["Number of Habitable Rooms"] = epcs["Number of Habitable Rooms"].astype(int) + epcs["Property Floor Area"] = epcs["Property Floor Area"].astype(float) + # Add additional columns as in your original code epcs["Estimated Number of Floors"] = epcs.apply( lambda x: estimate_number_of_floors(x["Property Type"]) if pd.notnull(x["Property Type"]) else None, axis=1 ) + epcs["Estimated Perimeter (m)"] = epcs.apply( lambda x: estimate_perimeter( x["Property Floor Area"] / x["Estimated Number of Floors"], @@ -167,7 +171,9 @@ def main(): """ # This should be set: - output_filepath = "/Users/khalimconn-kowlessar/Documents/hestia/Route Marches/PRS and OO properties - WC 11.11.2024" + output_filepath = ( + "/Users/khalimconn-kowlessar/Documents/hestia/Route Marches/PRS and OO properties - WC 11.11.2024.xlsx" + ) client = EpcClient(auth_token=EPC_AUTH_TOKEN) writer = pd.ExcelWriter(output_filepath, engine="xlsxwriter")