Update

Author: odow

src/Utilities/matrix_of_constraints.jl

@@ -290,11 +290,11 @@ If `S` is not part of the list, return `nothing`.
 function set_index end
 
 """
-    add_set(sets, i)::Int64
+    add_set(sets, i::Int)::Int64
 
 Add a scalar set of type index `i`.
 
-    add_set(sets, i, dim)::Int64
+    add_set(sets, i::Int, dim::Int)::Int64
 
 Add a vector set of type index `i` and dimension `dim`.
 
@@ -387,7 +387,7 @@ function MOI.get(
     return MOI.get(v.sets, attr)
 end
 
-_add_set(sets, i, ::MOI.AbstractScalarFunction) = add_set(sets, i)
+_add_set(sets, i::Int, ::MOI.AbstractScalarFunction) = add_set(sets, i)
 
 function _add_set(sets, i, func::MOI.AbstractVectorFunction)
     return add_set(sets, i, MOI.output_dimension(func))

src/Utilities/product_of_sets.jl

@@ -88,7 +88,7 @@ MOI.dimension(sets::MixOfScalarSets) = length(sets.set_ids)
 
 rows(::MixOfScalarSets, ci::MOI.ConstraintIndex) = ci.value
 
-function add_set(sets::MixOfScalarSets, i)
+function add_set(sets::MixOfScalarSets, i::Int)::Int64
     push!(sets.set_ids, i)
     return length(sets.set_ids)
 end
@@ -224,7 +224,7 @@ function rows(
     return sets.rows[i][ci.value]
 end
 
-function add_set(sets::OrderedProductOfSets, i, dim = 1)::Int
+function add_set(sets::OrderedProductOfSets, i::Int, dim::Int = 1)::Int64
     @assert !sets.final_touch
     push!(sets.rows[i], 1:dim)
     return length(sets.rows[i])