Codecov

Author: gdalle

DifferentiationInterface/ext/DifferentiationInterfaceMooncakeExt/utils.jl

@@ -22,7 +22,6 @@ nanify(x::AbstractFloat) = convert(typeof(x), NaN)
 nanify(x::AbstractArray) = map(nanify, x)
 nanify(x::NamedTuple) = NamedTuple{keys(x)}(map(nanify, values(x)))
 nanify(x::Tuple) = map(nanify, x)
-nanify(::NoFData) = NoFData()
 nanify(::NoRData) = NoRData()
 
 function nanify_fdata_and_rdata!!(contexts::Vararg{CoDual, C}) where {C}

DifferentiationInterface/src/misc/differentiate_with.jl

@@ -78,20 +78,8 @@ struct DifferentiateWith{C, F, B <: AbstractADType, N <: NTuple{C, Any}}
     function DifferentiateWith(
             f::F,
             backend::B,
-            context_wrappers::NTuple{C, Any},
+            context_wrappers::NTuple{C, <:Union{Function, Type}},
         ) where {F, B <: AbstractADType, C}
-        for (i, wrapper) in pairs(context_wrappers)
-            # Accept typical constructor-like values: functions or types.
-            if !(wrapper isa Function || wrapper isa Type)
-                throw(
-                    ArgumentError(
-                        "Each context wrapper must be a callable object or type " *
-                            "(e.g., a wrapper constructor like `Constant` or `Cache`), " *
-                            "but element $i has type $(typeof(wrapper)).",
-                    ),
-                )
-            end
-        end
         return new{C, F, B, typeof(context_wrappers)}(
             f,
             backend,

DifferentiationInterface/test/Back/DifferentiateWith/test.jl

@@ -127,3 +127,5 @@ end;
         ([2.0],),
     )
 end
+
+@test_throws MethodError DifferentiateWith(exp, AutoForwardDiff(), (3,))