Add rule for (not) generating tangents for vector spaces

Jutho · Jul 2, 2024 · b0e8230 · b0e8230
1 parent 7cc0d32
commit b0e8230
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/ad.jl b/test/ad.jl
@@ -15,6 +15,7 @@ end
 function ChainRulesTestUtils.rand_tangent(rng::AbstractRNG, x::AbstractTensorMap)
     return TensorMap(randn, scalartype(x), space(x))
 end
+ChainRulesTestUtils.rand_tangent(::AbstractRNG, ::VectorSpace) = NoTangent()
 function ChainRulesTestUtils.test_approx(actual::AbstractTensorMap,
                                          expected::AbstractTensorMap, msg=""; kwargs...)
     for (c, b) in blocks(actual)