Update MOI_wrapper.jl

jump-dev · Dec 3, 2023 · c7e7287 · c7e7287
1 parent abe4328
commit c7e7287
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/MOI/MOI_wrapper.jl b/src/MOI/MOI_wrapper.jl
@@ -527,15 +527,21 @@ function MOI.get(model::Optimizer, param::MOI.RawOptimizerAttribute)
     return
 end
 
-function MOI.set(model::Optimizer, ::MOI.TimeLimitSec, limit::Union{Real, Nothing})
-    if typeof(limit) == Real
-        MOI.set(model, MOI.RawOptimizerAttribute("CPXPARAM_TimeLimit"), limit)
-    end
+const _TIME_LIMIT_DEFAULT = 1.0e75
+
+function MOI.set(
+    model::Optimizer,
+    ::MOI.TimeLimitSec,
+    limit::Union{Nothing,Real},
+)
+    new_limit = something(limit, _TIME_LIMIT_DEFAULT)
+    MOI.set(model, MOI.RawOptimizerAttribute("CPXPARAM_TimeLimit"), limit)
     return
 end
 
 function MOI.get(model::Optimizer, ::MOI.TimeLimitSec)
-    return MOI.get(model, MOI.RawOptimizerAttribute("CPXPARAM_TimeLimit"))
+    limit = MOI.get(model, MOI.RawOptimizerAttribute("CPXPARAM_TimeLimit"))
+    return limit === _TIME_LIMIT_DEFAULT ? nothing : limit
 end
 
 MOI.supports_incremental_interface(::Optimizer) = true