diff --git a/Project.toml b/Project.toml index 1283bfc..f085158 100644 --- a/Project.toml +++ b/Project.toml @@ -14,7 +14,7 @@ MathOptSetDistances = "3b969827-a86c-476c-9527-bb6f1a8fbad5" SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf" [compat] -Dualization = "=0.6.1" +Dualization = "=0.7.0" MathOptSetDistances = "=0.2.11" [extras] diff --git a/src/layers/generic.jl b/src/layers/generic.jl index 661799e..cf79e03 100644 --- a/src/layers/generic.jl +++ b/src/layers/generic.jl @@ -60,12 +60,11 @@ function _make_completion_model(decomposition::AbstractDecomposition, dual_model filter_constraints=c -> !(c in y_cone_constraints) ) - # mark y and p as parameters (optimizing over z only) + # mark y as parameters (optimizing over z only) p_ref = getindex.(ref_map, get_p(dual_model, decomposition)) y_ref = getindex.(ref_map, get_y_dual(dual_model, decomposition)) y_ref_flat = reduce(vcat, y_ref) JuMP.@constraint(completion_model, y_ref_flat .∈ MOI.Parameter.(zeros(length(y_ref_flat)))) - JuMP.@constraint(completion_model, p_ref .∈ MOI.Parameter.(zeros(length(p_ref)))) return completion_model, (p_ref, y_ref, ref_map) end