remap p

klamike · klamike · commit 7f9fa01c7e18 · 2025-06-18T14:56:47.000-04:00
diff --git a/src/layers/generic.jl b/src/layers/generic.jl
@@ -61,6 +61,7 @@ function _make_completion_model(decomposition::AbstractDecomposition, dual_model
     )
 
     # 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))))

Original file line number	Diff line number	Diff line change
`@@ -61,6 +61,7 @@ function _make_completion_model(decomposition::AbstractDecomposition, dual_model`
`61`	`61`	`)`
`62`	`62`
`63`	`63`	`# mark y as parameters (optimizing over z only)`
	`64`	`+ p_ref = getindex.(ref_map, get_p(dual_model, decomposition))`
`64`	`65`	`y_ref = getindex.(ref_map, get_y_dual(dual_model, decomposition))`
`65`	`66`	`y_ref_flat = reduce(vcat, y_ref)`
`66`	`67`	`JuMP.@constraint(completion_model, y_ref_flat .∈ MOI.Parameter.(zeros(length(y_ref_flat))))`