From cf15052331af5b89678cb397ffa09b98924ffa2f Mon Sep 17 00:00:00 2001
From: troyyyyy <49091847+troyyyyy@users.noreply.github.com>
Date: Fri, 3 Mar 2023 16:48:46 +0800
Subject: [PATCH] Update utils.py

---
 abl/utils/utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/abl/utils/utils.py b/abl/utils/utils.py
index 05b97a4..fbb5bfb 100644
--- a/abl/utils/utils.py
+++ b/abl/utils/utils.py
@@ -76,9 +76,9 @@ def remapping_res(pred_res, m):
         remapping[value] = key
     return [[remapping[symbol] for symbol in formula] for formula in pred_res]
 
-def check_equal(a, b):
+def check_equal(a, b, max_err=0):
     if isinstance(a, (int, float)) and isinstance(b, (int, float)):
-        return abs(a - b) <= 1e-3
+        return abs(a - b) <= max_err
     
     if isinstance(a, list) and isinstance(b, list):
         if len(a) != len(b):
@@ -89,4 +89,4 @@ def check_equal(a, b):
         return True
     
     else:    
-        return a == b       
+        return a == b