From 46feb9e6bbd9eddebfbfaea053b35baa95d15d6f Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Fri, 17 May 2024 07:06:35 +0100 Subject: [PATCH] add previous assessors --- etl/epc_recommendations/find_my_epc_pipeline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/etl/epc_recommendations/find_my_epc_pipeline.py b/etl/epc_recommendations/find_my_epc_pipeline.py index 78a62903..a77ad44d 100644 --- a/etl/epc_recommendations/find_my_epc_pipeline.py +++ b/etl/epc_recommendations/find_my_epc_pipeline.py @@ -74,7 +74,7 @@ def retrieve_find_my_epc_data(uprn: int, postcode: str, address: str): other_certificates_section = address_res.find("div", {'id': 'other_certificates_and_reports'}) other_certificates_lines = other_certificates_section.find('dl', {"class": "govuk-summary-list"}) - other_certificates_links = other_certificates_lines.find_all('a') + other_certificates_links = other_certificates_lines.find_all('a') if other_certificates_lines is not None else [] if len(other_certificates_links) > 0: @@ -156,7 +156,7 @@ def main(): addresses_df['uprn'] = addresses_df['uprn'].astype(int) find_my_epc_data_list = [] - for i, row in tqdm(addresses_df.iterrows()): + for i, row in tqdm(addresses_df.tail(10).iterrows()): address_data = retrieve_find_my_epc_data( uprn=row['uprn'],