Update OptimizationDISparseExt.jl

Author: baggepinnen

src/OptimizationDISparseExt.jl

@@ -133,10 +133,10 @@ function instantiate_function(
         end
 
         function lagrangian(θ, σ, λ, p)
-            if iszero(θ)
-                return dot(λ, cons_oop(θ))
-            else
+            if eltype(θ) <: SparseConnectivityTracer.AbstractTracer || !iszero(θ)
                 return σ * f.f(θ, p) + dot(λ, cons_oop(θ))
+            else
+                return dot(λ, cons_oop(θ))
             end
         end
     end