From 064b202e854bfb42a0f972452a879ff0d4e2ec4e Mon Sep 17 00:00:00 2001 From: wout4 Date: Thu, 21 Sep 2023 10:19:42 +0200 Subject: [PATCH] import --- cpmpy/transformations/normalize.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/cpmpy/transformations/normalize.py b/cpmpy/transformations/normalize.py index fcff3b82d..701854d68 100644 --- a/cpmpy/transformations/normalize.py +++ b/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 \ No newline at end of file + 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) \ No newline at end of file