Fix Indicator constraints in ListOfConstraintTypesPresent (#454)

odow · web-flow · commit 6cc6dd72942a · 2025-04-03T11:09:20.000+13:00
diff --git a/src/MOI/MOI_wrapper.jl b/src/MOI/MOI_wrapper.jl
@@ -326,6 +326,7 @@ function MOI.empty!(model::Optimizer)
     empty!(model.affine_constraint_info)
     empty!(model.quadratic_constraint_info)
     empty!(model.sos_constraint_info)
+    empty!(model.indicator_constraint_info)
     model.name_to_variable = nothing
     model.name_to_constraint_index = nothing
     model.ret_optimize = Cint(0)
@@ -3586,6 +3587,12 @@ function MOI.get(model::Optimizer, ::MOI.ListOfConstraintTypesPresent)
     for info in values(model.sos_constraint_info)
         push!(constraints, (MOI.VectorOfVariables, typeof(info.set)))
     end
+    for (info, _) in values(model.indicator_constraint_info)
+        push!(
+            constraints,
+            (MOI.VectorAffineFunction{Float64}, typeof(info.set)),
+        )
+    end
     return collect(constraints)
 end
 
diff --git a/test/MathOptInterface/MOI_wrapper.jl b/test/MathOptInterface/MOI_wrapper.jl
@@ -313,6 +313,22 @@ function test_example_biobjective_knapsack()
     return
 end
 
+function test_ListOfConstraintTypesPresent_indicator()
+    model = CPLEX.Optimizer()
+    x = MOI.add_variable(model)
+    z = MOI.add_variable(model)
+    MOI.add_constraint(model, z, MOI.ZeroOne())
+    c = MOI.add_constraint(
+        model,
+        MOI.Utilities.operate(vcat, Float64, 1.0 * z, 1.0 * x),
+        MOI.Indicator{MOI.ACTIVATE_ON_ONE}(MOI.EqualTo(1.0)),
+    )
+    F = MOI.VectorAffineFunction{Float64}
+    S = MOI.Indicator{MOI.ACTIVATE_ON_ONE,MOI.EqualTo{Float64}}
+    @test (F, S) in MOI.get(model, MOI.ListOfConstraintTypesPresent())
+    return
+end
+
 end  # module TestMOIwrapper
 
 TestMOIwrapper.runtests()
