From 55a56bb7062910fb1aedb75f57b71dc7dc62abcb Mon Sep 17 00:00:00 2001 From: troyyyyy Date: Fri, 10 Nov 2023 12:35:19 +0800 Subject: [PATCH] [FIX] add bool check in _check_equal --- abl/reasoning/kb.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/abl/reasoning/kb.py b/abl/reasoning/kb.py index 0fc9a5c..63a4d29 100644 --- a/abl/reasoning/kb.py +++ b/abl/reasoning/kb.py @@ -88,7 +88,11 @@ class KBBase(ABC): """ if logic_result == None: return False - return abs(logic_result - y) <= self.max_err + + if isinstance(logic_result, (int, float)) and isinstance(y, (int, float)): + return abs(logic_result - y) <= self.max_err + else: + return logic_result == y def revise_at_idx(self, pred_pseudo_label, y, revision_idx): """