adds supports_variablerates trait.

SciML · Dec 11, 2022 · 27707e6 · 27707e6
1 parent b87e00e
commit 27707e6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/aggregators/aggregators.jl b/src/aggregators/aggregators.jl
@@ -178,6 +178,10 @@ needs_vartojumps_map(aggregator::AbstractAggregatorAlgorithm) = false
 needs_vartojumps_map(aggregator::RSSA) = true
 needs_vartojumps_map(aggregator::RSSACR) = true
 
+# true if aggregator supports variable rates
+supports_variablerates(aggregator::AbstractAggregatorAlgorithm) = false
+supports_variablerates(aggregator::Coevolve) = true
+
 is_spatial(aggregator::AbstractAggregatorAlgorithm) = false
 is_spatial(aggregator::NSM) = true
 is_spatial(aggregator::DirectCRDirect) = true
diff --git a/src/problem.jl b/src/problem.jl
@@ -197,7 +197,7 @@ function JumpProblem(prob, aggregator::AbstractAggregatorAlgorithm, jumps::JumpS
         cont_agg = JumpSet().variable_jumps
         disc_agg = nothing
         constant_jump_callback = CallbackSet()
-    elseif typeof(aggregator) <: Coevolve
+    elseif supports_variablerates(aggregator)
         # Coevolve handles all types of jumps together
         variable_jumps = VariableRateJump[]
         if (length(jumps.constant_jumps) == 0)