compact checks

Author: isaacsas

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