Pyomo从6.5.0升级至6.6.2后布尔值与表达式相乘报错求助

问题原因 #

Pyomo 6.6.x版本对表达式（Expression）的类型系统进行了严格化调整，移除了对Python原生bool类型与各类Expression对象（如_GeneralExpressionData、SumExpression等）乘法运算的隐式支持。而Pyomo的二进制变量（Binary Var）属于框架内定义的类型，保留了对应的运算符重载逻辑，因此可以正常执行乘法操作。

高效解决方案 #

方案1：全局运算符补丁（无需修改业务代码） #

通过给Pyomo核心Expression相关类型添加bool乘法支持，一次性解决全项目的兼容问题。在项目初始化的入口代码中添加以下补丁：

import pyomo.core.expr as expr

# 为_GeneralExpressionData添加bool乘法支持
def _gen_expr_mul_bool(self, other):
    if isinstance(other, bool):
        return self * (1 if other else 0)
    return NotImplemented

def _gen_expr_rmul_bool(self, other):
    if isinstance(other, bool):
        return (1 if other else 0) * self
    return NotImplemented

expr._GeneralExpressionData.__mul__ = _gen_expr_mul_bool
expr._GeneralExpressionData.__rmul__ = _gen_expr_rmul_bool

# 为SumExpression添加bool乘法支持
def _sum_expr_mul_bool(self, other):
    if isinstance(other, bool):
        return self * (1 if other else 0)
    return NotImplemented

def _sum_expr_rmul_bool(self, other):
    if isinstance(other, bool):
        return (1 if other else 0) * self
    return NotImplemented

expr.SumExpression.__mul__ = _sum_expr_mul_bool
expr.SumExpression.__rmul__ = _sum_expr_rmul_bool

# 为ScalarExpression添加bool乘法支持
def _scalar_expr_mul_bool(self, other):
    if isinstance(other, bool):
        return self * (1 if other else 0)
    return NotImplemented

def _scalar_expr_rmul_bool(self, other):
    if isinstance(other, bool):
        return (1 if other else 0) * self
    return NotImplemented

expr.ScalarExpression.__mul__ = _scalar_expr_mul_bool
expr.ScalarExpression.__rmul__ = _scalar_expr_rmul_bool

# 为NPV_SumExpression添加bool乘法支持（按需添加）
if hasattr(expr, 'NPV_SumExpression'):
    def _npv_sum_expr_mul_bool(self, other):
        if isinstance(other, bool):
            return self * (1 if other else 0)
        return NotImplemented

    def _npv_sum_expr_rmul_bool(self, other):
        if isinstance(other, bool):
            return (1 if other else 0) * self
        return NotImplemented

    expr.NPV_SumExpression.__mul__ = _npv_sum_expr_mul_bool
    expr.NPV_SumExpression.__rmul__ = _npv_sum_expr_rmul_bool

该补丁会自动将参与乘法的布尔值转换为整数1/0，完全兼容原有业务代码的写法，无需任何修改。

方案2：局部工具函数（适合小范围调整） #

如果不想全局修改框架行为，可以编写一个工具函数，将布尔值转换为Pyomo的Param对象（Param与Expression的乘法不受版本限制）：

import pyomo.environ as pyo

def bool_to_param(b: bool):
    return pyo.Param(initialize=1 if b else 0)

使用时替换原有布尔值即可：

# 原代码：attribute[index] * add_expression
# 修改后：attribute[index] * bool_to_param(add_expression)

方案3：修改Pyomo源码（不推荐） #

直接修改Pyomo安装目录中对应表达式类型的运算符重载方法，添加bool类型处理逻辑。但此方法会影响所有使用该Pyomo环境的项目，且Pyomo版本升级后修改会被覆盖，仅作为临时应急方案。

验证示例 #

将方案1的补丁代码添加到你的工作示例最开头，运行后即可正常执行，无需修改update_expression函数中的add_expression参数。

内容的提问来源于stack exchange，提问作者djo