diff --git a/cpmpy/transformations/flatten_model.py b/cpmpy/transformations/flatten_model.py index b5d936b29..8848a001f 100644 --- 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):