OR-Tools中LinearExpr作布尔值判断抛出NotImplementedError问题求助

问题复现 #

开发排班算法时，需统计特定员工每日的排班班次总数，代码运行至布尔条件判断环节时报错终止，原始实现代码如下：

for e in self.all_employees:
    for sd in start_day:
        if self.employee_leaves[(e,sd)] == 1:
            sum1 = sum(self.shifts[e,sd,s] for s in self.all_shifts)

运行后抛出NotImplementedError异常，具体报错信息：

NotImplementedError: Evaluating a BoundedLinearExpression '((shift_n0d14s0 + shift_n0d14s1) + shift_n0d14s2) == 0' as a Boolean value is not supported.

报错核心信息为：不支持将BoundedLinearExpression（有界线性表达式）直接作为布尔值求值，对应逻辑无法执行。

报错根因 #

该问题常见于使用OR-Tools等整数规划求解器搭建排班模型的场景：
代码中self.employee_leaves、self.shifts存储的不是Python原生的整数、布尔值，而是求解器自定义的线性表达式对象。Python原生if语法会自动将括号内的内容转换为布尔类型，但求解器没有实现这类表达式直接转布尔值的逻辑，因此抛出异常。
本质原因是模型构建阶段，所有决策变量还未完成求解，本身不存在确定的数值，自然无法直接做布尔判断。

修复方案 #

根据代码所处阶段选择对应写法即可：

• 若处于模型约束构建阶段：不要用Python原生if写条件逻辑，调用求解器自带的条件约束API实现规则。比如要实现「员工当日请假则当日排班总次数为0」的规则，写法参考：

for e in self.all_employees:
    for sd in start_day:
        daily_shift_total = sum(self.shifts[e,sd,s] for s in self.all_shifts)
        # 调用求解器自带的条件约束接口，请假时触发当日班次总数为0的约束
        self.model.Add(daily_shift_total == 0).OnlyEnforceIf(self.employee_leaves[(e,sd)] == 1)

• 若处于求解完成后的结果统计阶段：先通过求解器的取值接口拿到变量的实际计算结果，再做布尔判断和数值统计，写法参考：

# 需提前完成求解，拿到solver实例与合法求解状态
for e in self.all_employees:
    for sd in start_day:
        # 先取出变量的实际求解值，再做判断
        if self.solver.Value(self.employee_leaves[(e,sd)]) == 1:
            sum1 = sum(self.solver.Value(self.shifts[e,sd,s]) for s in self.all_shifts)

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