From 42a43b3cca4d5ebf85efcdb02990de136a58ce86 Mon Sep 17 00:00:00 2001
From: Tony-HYX <605698554@qq.com>
Date: Mon, 6 Mar 2023 16:24:28 +0800
Subject: [PATCH] update para of method

---
 abl/abducer/kb.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/abl/abducer/kb.py b/abl/abducer/kb.py
index c5fc675..632bd42 100644
--- a/abl/abducer/kb.py
+++ b/abl/abducer/kb.py
@@ -24,6 +24,7 @@ from multiprocessing import Pool
 import pyswip
 
 class KBBase(ABC):
+    # TODO：有些不能是默认参数，必须给定
     def __init__(self, pseudo_label_list=None, len_list=None, GKB_flag=False, max_err=0):
         self.pseudo_label_list = pseudo_label_list
         self.len_list = len_list
@@ -70,7 +71,7 @@ class KBBase(ABC):
         return X, Y
 
     @abstractmethod
-    def logic_forward(self):
+    def logic_forward(self, pseudo_labels):
         pass
     
     def _logic_forward(self, xs, multiple_predictions=False):
@@ -87,7 +88,7 @@ class KBBase(ABC):
             return self._abduce_by_search(pred_res, key, max_address_num, require_more_address, multiple_predictions)
     
     @abstractmethod
-    def _find_candidate_GKB(self):
+    def _find_candidate_GKB(self, pred_res, key):
         pass
     
     def _abduce_by_GKB(self, pred_res, key, max_address_num, require_more_address, multiple_predictions):
@@ -260,7 +261,7 @@ class prolog_KB(KBBase):
         query_string += ",%s)." % key if not key_is_none_flag else ")."
         return query_string
 
-    def _find_candidate_GKB(self):
+    def _find_candidate_GKB(self, pred_res, key):
         pass
     
     def address_by_idx(self, pred_res, key, address_idx, multiple_predictions=False):