From 7640e2ab0c61efe57aa4e3ff7a32dea38f3e9a21 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Thu, 14 Jul 2022 21:38:00 +0400 Subject: [PATCH] Wrap attempt to extract dates from entry in try/catch - Not all YYYY-MM-DD strings in entry are necessarily dates --- src/search_filter/date_filter.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/search_filter/date_filter.py b/src/search_filter/date_filter.py index 578366ff..39fd28ae 100644 --- a/src/search_filter/date_filter.py +++ b/src/search_filter/date_filter.py @@ -36,7 +36,10 @@ def date_filter(query, entries, embeddings): # Extract dates from entry for date_in_entry_string in re.findall(r'\d{4}-\d{2}-\d{2}', entry[1]): # Convert date string in entry to unix timestamp - date_in_entry = datetime.strptime(date_in_entry_string, '%Y-%m-%d').timestamp() + try: + date_in_entry = datetime.strptime(date_in_entry_string, '%Y-%m-%d').timestamp() + except ValueError: + continue # Check if date in entry is within date range specified in query if query_daterange[0] <= date_in_entry < query_daterange[1]: entries_to_include.add(id)