From c5fadfcbb69f1c52cc5926312def088e396cfd5f Mon Sep 17 00:00:00 2001
From: troyyyyy <huwc@lamda.nju.edu.cn>
Date: Tue, 14 Mar 2023 15:05:48 +0800
Subject: [PATCH] Rearrange rules to

---
 abl/abducer/kb.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/abl/abducer/kb.py b/abl/abducer/kb.py
index 333901e..feba8c4 100644
--- a/abl/abducer/kb.py
+++ b/abl/abducer/kb.py
@@ -271,12 +271,16 @@ class prolog_KB(KBBase):
             candidates.append(candidate)
         return candidates
 
+
+class HED_prolog_KB(prolog_KB):
+    def __init__(self, pseudo_label_list, pl_file):
+        super().__init__(pseudo_label_list, pl_file)
+        
     def consist_rule(self, exs, rules):
         rules = str(rules).replace("\'","")
         return len(list(self.prolog.query("eval_inst_feature(%s, %s)." % (exs, rules)))) != 0
 
     def abduce_rules(self, pred_res):
-        # print(pred_res)
         prolog_result = list(self.prolog.query("consistent_inst_feature(%s, X)." % pred_res))
         if len(prolog_result) == 0:
             return None