[POC] Allow array in nonlinear expressions

jump-dev · Aug 15, 2023 · c91039e · c91039e
1 parent ef2b488
commit c91039e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/nlp_expr.jl b/src/nlp_expr.jl
@@ -294,6 +294,10 @@ for f in MOI.Nonlinear.DEFAULT_UNIVARIATE_OPERATORS
     end
 end
 
+function LinearAlgebra.det(A::LinearAlgebra.Symmetric{<:AbstractJuMPScalar})
+    return GenericNonlinearExpr{variable_ref_type(eltype(A))}(:det, A)
+end
+
 # Multivariate operators
 
 # The multivariate operators in MOI are +, -, *, ^, /, ifelse, atan
@@ -460,6 +464,8 @@ function moi_function(f::GenericNonlinearExpr)
             end
         elseif arg isa Number
             push!(parent.args, arg)
+        elseif arg isa AbstractArray
+            push!(parent.args, moi_function.(arg))
         else
             push!(parent.args, moi_function(arg))
         end
@@ -484,6 +490,8 @@ function jump_function(model::GenericModel, f::MOI.ScalarNonlinearFunction)
             end
         elseif arg isa Number
             push!(parent.args, arg)
+        elseif arg isa AbstractArray
+            push!(parent.args, jump_function.(model, arg))
         else
             push!(parent.args, jump_function(model, arg))
         end