Fixes

Author: gdalle

DifferentiationInterface/ext/DifferentiationInterfaceSparseMatrixColoringsExt/hessian.jl

@@ -79,11 +79,11 @@ end
 function DI.hessian!(
     f::F,
     hess,
-    prep::SparseHessianPrep{<:DI.BatchSizeSettings{B}},
+    prep::SparseHessianPrep{SIG,<:DI.BatchSizeSettings{B}},
     backend::AutoSparse,
     x,
     contexts::Vararg{DI.Context,C},
-) where {F,B,C}
+) where {F,SIG,B,C}
     DI.check_prep(f, prep, backend, x, contexts...)
     (;
         batch_size_settings,
@@ -124,8 +124,8 @@ function DI.hessian!(
 end
 
 function DI.hessian(
-    f::F, prep::SparseHessianPrep{B}, backend::AutoSparse, x, contexts::Vararg{DI.Context,C}
-) where {F,B,C}
+    f::F, prep::SparseHessianPrep, backend::AutoSparse, x, contexts::Vararg{DI.Context,C}
+) where {F,C}
     DI.check_prep(f, prep, backend, x, contexts...)
     hess = similar(sparsity_pattern(prep), eltype(x))
     return DI.hessian!(f, hess, prep, backend, x, contexts...)

DifferentiationInterface/ext/DifferentiationInterfaceSparseMatrixColoringsExt/jacobian.jl

@@ -260,11 +260,11 @@ end
 function _sparse_jacobian_aux!(
     f_or_f!y::FY,
     jac,
-    prep::PushforwardSparseJacobianPrep{<:DI.BatchSizeSettings{B}},
+    prep::PushforwardSparseJacobianPrep{SIG,<:DI.BatchSizeSettings{B}},
     backend::AutoSparse,
     x,
     contexts::Vararg{DI.Context,C},
-) where {FY,B,C}
+) where {FY,SIG,B,C}
     (;
         batch_size_settings,
         coloring_result,
@@ -306,11 +306,11 @@ end
 function _sparse_jacobian_aux!(
     f_or_f!y::FY,
     jac,
-    prep::PullbackSparseJacobianPrep{<:DI.BatchSizeSettings{B}},
+    prep::PullbackSparseJacobianPrep{SIG,<:DI.BatchSizeSettings{B}},
     backend::AutoSparse,
     x,
     contexts::Vararg{DI.Context,C},
-) where {FY,B,C}
+) where {FY,SIG,B,C}
     (;
         batch_size_settings,
         coloring_result,

DifferentiationInterface/ext/DifferentiationInterfaceSparseMatrixColoringsExt/jacobian_mixed.jl

@@ -164,12 +164,12 @@ function _sparse_jacobian_aux!(
     f_or_f!y::FY,
     jac,
     prep::MixedModeSparseJacobianPrep{
-        <:DI.BatchSizeSettings{Bf},<:DI.BatchSizeSettings{Br}
+        SIG,<:DI.BatchSizeSettings{Bf},<:DI.BatchSizeSettings{Br}
     },
     backend::AutoSparse,
     x,
     contexts::Vararg{DI.Context,C},
-) where {FY,Bf,Br,C}
+) where {FY,SIG,Bf,Br,C}
     (;
         batch_size_settings_forward,
         batch_size_settings_reverse,