From 60878625214b96114cb5a4f747071b6079e017e0 Mon Sep 17 00:00:00 2001
From: Debanjum Singh Solanky <debanjum@gmail.com>
Date: Sun, 4 Sep 2022 16:42:28 +0300
Subject: [PATCH] Use LRU helper class for explicit filter cache

---
 src/search_filter/explicit_filter.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/search_filter/explicit_filter.py b/src/search_filter/explicit_filter.py
index 6f64ede5..7a26f830 100644
--- a/src/search_filter/explicit_filter.py
+++ b/src/search_filter/explicit_filter.py
@@ -3,13 +3,12 @@ import re
 import time
 import pickle
 import logging
-from copy import deepcopy
 
 # External Packages
 import torch
 
 # Internal Packages
-from src.utils.helpers import resolve_absolute_path
+from src.utils.helpers import LRU, resolve_absolute_path
 from src.utils.config import SearchType
 
 
@@ -26,7 +25,7 @@ class ExplicitFilter:
         self.entry_key = entry_key
         self.search_type = search_type
         self.word_to_entry_index = dict()
-        self.cache = {}
+        self.cache = LRU()
 
 
     def load(self, entries, regenerate=False):