compact checks

SciML · Jan 5, 2023 · 32cda2c · 32cda2c
1 parent 640eced
commit 32cda2c
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/problem.jl b/src/problem.jl
@@ -187,17 +187,14 @@ function JumpProblem(prob, aggregator::AbstractAggregatorAlgorithm, jumps::JumpS
     if spatial_system !== nothing && hopping_constants !== nothing
         (num_crjs(jumps) == num_vrjs(jumps) == 0) ||
             error("Spatial aggregators only support MassActionJumps currently.")
-        if !is_spatial(aggregator)
+
+        if is_spatial(aggregator)
+            kwargs = merge((; hopping_constants, spatial_system), kwargs)
+        else
             prob, maj = flatten(maj, prob, spatial_system, hopping_constants; kwargs...)
         end
     end
 
-    if is_spatial(aggregator)
-        (num_crjs(jumps) == num_vrjs(jumps) == 0) ||
-            error("Spatial aggregators only support MassActionJumps currently.")
-        kwargs = merge((; hopping_constants, spatial_system), kwargs)
-    end
-
     ndiscjumps = get_num_majumps(maj) + num_crjs(jumps)
 
     # separate bounded variable rate jumps *if* the aggregator can use them