Fix test

FluxML · Jan 18, 2025 · e68a4ad · e68a4ad
1 parent 8baa2b2
commit e68a4ad
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/chainrules.jl b/test/chainrules.jl
@@ -445,12 +445,12 @@ end
 end
 
 @testset "No thunks in the gradient" begin
-    struct Dense
+    struct CustomDense
         w::Matrix{Float32}
     end
-    (d::Dense)(x) = d.w * x
+    (d::CustomDense)(x) = d.w * x
 
-    layers = [Dense(rand(Float32, 3, 3))]
+    layers = [CustomDense(rand(Float32, 3, 3))]
     x = ones(Float32, 3)
     g = gradient(layers -> sum(layers[1](x)), layers)[1]
     @test g[1] isa NamedTuple