Support constant objectives in JuMPExtension

jump-dev · Apr 8, 2019 · 62449e3 · 62449e3
1 parent 587f8ed
commit 62449e3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/objective.jl b/src/objective.jl
@@ -46,9 +46,9 @@ function set_objective_sense(model::Model, sense::MOI.OptimizationSense)
 end
 
 """
-    set_objective_function(model::Model,
-                           func::Union{AbstractJuMPScalar,
-                                       MathOptInterface.AbstractScalarFunction})
+    set_objective_function(
+        model::Model,
+        func::Union{AbstractJuMPScalar, MathOptInterface.AbstractScalarFunction})
 
 Sets the objective function of the model to the given function. See
 [`set_objective_sense`](@ref) to set the objective sense. These are low-level

diff --git a/test/JuMPExtension.jl b/test/JuMPExtension.jl
@@ -230,6 +230,10 @@ function JuMP.set_objective(m::MyModel, sense::MOI.OptimizationSense,
     m.objectivesense = sense
     m.objective_function = f
 end
+function JuMP.set_objective(m::MyModel, sense::MOI.OptimizationSense, f::Real)
+    m.objectivesense = sense
+    m.objective_function = JuMP.GenericAffExpr{Float64, MyVariableRef}(f)
+end
 JuMP.objective_sense(model::MyModel) = model.objectivesense
 function JuMP.set_objective_sense(model::MyModel, sense)
     model.objectivesense = sense