import

cpmpy/transformations/normalize.py

@@ -3,8 +3,6 @@
 import numpy as np
 import builtins
 
-from cpmpy.transformations.flatten_model import __is_flat_var
-
 from ..expressions.core import BoolVal, Expression, Comparison, Operator
 from ..expressions.utils import eval_comparison, is_false_cst, is_true_cst, is_boolexpr, is_num
 from ..expressions.variables import NDVarArray
@@ -252,4 +250,10 @@ def normalize_boolexpr(lst_of_expr):
                         continue
 
         newlist.append(expr)
-    return newlist
+    return newlist
+
+
+def __is_flat_var(arg):
+    """ True if the variable is a numeric constant, or a _NumVarImpl (incl subclasses)
+    """
+    return is_num(arg) or isinstance(arg, _NumVarImpl)