Update Dualization.jl (#10)

* update dualization

* bump

* remap p

Author: klamike

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]

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