From c38787969441fc18f6b7403e35760aa8c79d74bb Mon Sep 17 00:00:00 2001
From: troyyyyy <huwc@lamda.nju.edu.cn>
Date: Wed, 15 Nov 2023 14:12:28 +0800
Subject: [PATCH] [MNT] support abl_cache

---
 abl/reasoning/kb.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/abl/reasoning/kb.py b/abl/reasoning/kb.py
index b626504..e1df939 100644
--- a/abl/reasoning/kb.py
+++ b/abl/reasoning/kb.py
@@ -10,7 +10,7 @@ import numpy as np
 import pyswip
 
 from ..utils.utils import flatten, reform_idx, hamming_dist, to_hashable, restore_from_hashable
-
+from ..utils.cache import abl_cache
 
 class KBBase(ABC):
     """
@@ -179,7 +179,7 @@ class KBBase(ABC):
             candidates.extend(self._revision(revision_num, pred_pseudo_label, y))
         return candidates
     
-    @lru_cache(maxsize=4096)
+    @abl_cache(max_size=4096)
     def _abduce_by_search_cache(self, pred_pseudo_label, y, max_revision_num, require_more_revision):
         """
         `_abduce_by_search` with cache.