From 2160df0bb79ccc20f2f484bc1a29af07105fc3d3 Mon Sep 17 00:00:00 2001
From: troyyyyy <49091847+troyyyyy@users.noreply.github.com>
Date: Mon, 6 Mar 2023 13:57:17 +0800
Subject: [PATCH] Update kb.py

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

diff --git a/abl/abducer/kb.py b/abl/abducer/kb.py
index 7aff683..71bdcbe 100644
--- a/abl/abducer/kb.py
+++ b/abl/abducer/kb.py
@@ -282,9 +282,7 @@ class prolog_KB(KBBase):
 class add_prolog_KB(prolog_KB):
     def __init__(self, pseudo_label_list=list(range(10))):
         super().__init__(pseudo_label_list)
-        for i in self.pseudo_label_list:
-            self.prolog.assertz("pseudo_label(%s)" % i)
-        self.prolog.assertz("addition(Z1, Z2, Res) :- pseudo_label(Z1), pseudo_label(Z2), Res is Z1+Z2")
+        self.prolog.consult('../datasets/mnist_add/add.pl')
 
 
 class HED_prolog_KB(prolog_KB):