Fix coverage

Author: gdalle

DifferentiationInterface/ext/DifferentiationInterfaceMooncakeExt/DifferentiationInterfaceMooncakeExt.jl

@@ -33,7 +33,7 @@ using Mooncake:
 
 const AnyAutoMooncake{C} = Union{AutoMooncake{C},AutoMooncakeForward{C}}
 
-DI.check_available(::AnyAutoMooncake) = true
+DI.check_available(::AnyAutoMooncake{C}) where {C} = true
 
 get_config(::AnyAutoMooncake{Nothing}) = Config()
 get_config(backend::AnyAutoMooncake{<:Config}) = backend.config

DifferentiationInterface/ext/DifferentiationInterfaceMooncakeExt/forward_twoarg.jl

@@ -58,7 +58,7 @@ end
 function DI.pushforward(
     f!::F,
     y,
-    prep::MooncakeOneArgPushforwardPrep,
+    prep::MooncakeTwoArgPushforwardPrep,
     backend::AutoMooncakeForward,
     x,
     tx::NTuple,
@@ -72,7 +72,7 @@ function DI.value_and_pushforward!(
     f!::F,
     y::Y,
     ty::NTuple,
-    prep::MooncakeOneArgPushforwardPrep,
+    prep::MooncakeTwoArgPushforwardPrep,
     backend::AutoMooncakeForward,
     x::X,
     tx::NTuple,
@@ -100,7 +100,7 @@ function DI.pushforward!(
     f!::F,
     y,
     ty::NTuple,
-    prep::MooncakeOneArgPushforwardPrep,
+    prep::MooncakeTwoArgPushforwardPrep,
     backend::AutoMooncakeForward,
     x,
     tx::NTuple,