rerun tests

SciML · Jan 7, 2024 · 39bd9c7 · 39bd9c7
1 parent 2e57081
commit 39bd9c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/loss_function_generation.jl b/src/loss_function_generation.jl
@@ -166,7 +166,7 @@ function generate_derivative_rules(term, eqdata, eltypeθ, dummyvars, derivative
         end
     end
     vr = mapreduce(vcat, dvs, init = []) do w
-        @rule w => ufunc(w, phi, varmap)(reducevcat(arguments(w), eltypeθ), θ)
+        @rule w => ufunc(w, phi, varmap)(reducevcat(arguments(w), eltypeθ, switch), θ)
     end
 
     sr = @rule switch => 1