diff --git a/applications/modelling_e2e/handler.py b/applications/modelling_e2e/handler.py index 6b276f0c6..7c8bc4f99 100644 --- a/applications/modelling_e2e/handler.py +++ b/applications/modelling_e2e/handler.py @@ -46,6 +46,7 @@ from repositories.geospatial.geospatial_s3_repository import ( GeospatialS3Repository, ParquetReader, ) +from repositories.epc.epc_postgres_repository import EpcPostgresRepository from repositories.plan.plan_postgres_repository import PlanPostgresRepository from repositories.product.product_postgres_repository import ProductPostgresRepository from repositories.property.landlord_override_overlays import overlays_from @@ -194,6 +195,9 @@ def handler(body: dict[str, Any], context: Any) -> None: measure_types = ", ".join(m.measure_type for m in plan.measures) or "none" logger.info(f"[dry_run] measures=[{measure_types}] — skipping DB write") return + EpcPostgresRepository(session).save( + epc, property_id=property_id, portfolio_id=portfolio_id + ) PlanPostgresRepository(session).save( plan, property_id=property_id, diff --git a/applications/modelling_e2e/local_handler/invoke_local_lambda.py b/applications/modelling_e2e/local_handler/invoke_local_lambda.py index 808200c62..e28d6e845 100644 --- a/applications/modelling_e2e/local_handler/invoke_local_lambda.py +++ b/applications/modelling_e2e/local_handler/invoke_local_lambda.py @@ -12,11 +12,11 @@ payload = { { "body": json.dumps( { - "property_id": 709634, - "portfolio_id": 785, - "scenario_id": 1266, + "property_id": 709774, + "portfolio_id": 796, + "scenario_id": 1268, "no_solar": False, - "dry_run": True, + "dry_run": False, } ) }