From db7244d2be0a7b31e7c4cdfb07e3acf7e88b31e6 Mon Sep 17 00:00:00 2001 From: troyyyyy <49091847+troyyyyy@users.noreply.github.com> Date: Tue, 15 Nov 2022 20:25:54 +0800 Subject: [PATCH] Update kb.py --- abducer/kb.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/abducer/kb.py b/abducer/kb.py index 9c09765..b60f50e 100644 --- a/abducer/kb.py +++ b/abducer/kb.py @@ -99,8 +99,8 @@ class add_KB(KBBase): address_idx_list = list(combinations(list(range(len(pred_res))), address_num)) for address_idx in address_idx_list: for c in all_address_candidate: - if(np.count_nonzero(np.array(c) != np.array(pred_res)[np.array(address_idx)]) == address_num): - pred_res_array = np.array(pred_res) + pred_res_array = np.array(pred_res) + if(np.count_nonzero(np.array(c) != pred_res_array[np.array(address_idx)]) == address_num): pred_res_array[np.array(address_idx)] = c if(self.logic_forward(pred_res_array) == key): candidates.append(pred_res_array) @@ -116,8 +116,8 @@ class add_KB(KBBase): address_idx_list = list(combinations(list(range(len(pred_res))), address_num)) for address_idx in address_idx_list: for c in all_candidate: - if(np.count_nonzero(np.array(c) != np.array(pred_res)[np.array(address_idx)]) == address_num): - pred_res_array = np.array(pred_res) + pred_res_array = np.array(pred_res) + if(np.count_nonzero(np.array(c) != pred_res_array[np.array(address_idx)]) == address_num): pred_res_array[np.array(address_idx)] = c if(self.logic_forward(pred_res_array) == key): candidates.append(pred_res_array)