use is_boolexpr

CPMpy · Jan 12, 2024 · 2452dac · 2452dac
1 parent c73d21c
commit 2452dac
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cpmpy/transformations/flatten_model.py b/cpmpy/transformations/flatten_model.py
@@ -308,9 +308,10 @@ def get_or_make_var(expr,boolean=False):
         Determines whether this is a Boolean or Integer variable and returns
         the equivalent of: (var, normalize(expr) == var)
     """
-    if is_boolexpr(expr):
+    isbool = is_boolexpr(expr)
+    if isbool:
         boolean = True
-    if __is_flat_var(expr) and boolean == expr.is_bool():
+    if __is_flat_var(expr) and boolean == isbool:
         return (expr, [])
 
     if is_any_list(expr):