From 14452dde9937d242c30ff490b8c5039a80ea6fcc Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Tue, 28 May 2024 19:07:58 +0100 Subject: [PATCH] use pandas --- etl/epc/generate_scenarios_data.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/etl/epc/generate_scenarios_data.py b/etl/epc/generate_scenarios_data.py index f9f66034..df1f9452 100644 --- a/etl/epc/generate_scenarios_data.py +++ b/etl/epc/generate_scenarios_data.py @@ -41,6 +41,15 @@ cleaning_data = read_dataframe_from_s3_parquet( materials = get_materials(session) cleaned = get_cleaned() +# TODO: THIS IS A TEMPORARY FIX +new_walls_description_mapping = pd.DataFrame(cleaned["walls-description"]) +new_walls_description_mapping.loc[ + ~new_walls_description_mapping["thermal_transmittance_unit"].isnull(), + "thermal_transmittance_unit", +] = "w/m-¦k" + +cleaned["walls-description"] = new_walls_description_mapping.to_dict(orient="records") + uprn_filenames = read_dataframe_from_s3_parquet( bucket_name=get_settings().DATA_BUCKET, file_key="spatial/filename_meta.parquet" ) @@ -167,7 +176,7 @@ for scenario_property in scenario_properties: p.get_components(cleaned, photo_supply_lookup, floor_area_decile_thresholds) recommender = Recommendations(property_instance=p, materials=materials) - property_recommendations = recommender.recommend("0") + property_recommendations = recommender.recommend() wall_recommendations = recommender.wall_recomender.recommendations loft_recommendations = recommender.roof_recommender.recommendations