Correct tests re #191

invenia · Jul 2, 2021 · 5369e2d · 5369e2d
1 parent fb45de7
commit 5369e2d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/sensitivities/functional/reducedim.jl b/test/sensitivities/functional/reducedim.jl
@@ -54,8 +54,8 @@
                          randn(rng, 10, 10, 10), randn(rng, 10, 10, 10))
 
         # Issue #123
-        x6_ = collect(1:10)
-        tens = (fill(10.0, (10,)), fill(10.0, (10, 1)))
+        x6_ = float.(1:10)
+        tens = (fill(10.0, (10,)), fill(10.0, (10,)))
         @test ∇(x->sum(sum(x, dims=2)))(x6_) == (oneslike(x6_),)
         @test ∇((x, y)->sum(sum(x, dims=2) .+ sum(y, dims=2)'))(x6_, x6_) == tens
         @test ∇((x, y)->sum(x .+ y'))(x6_, x6_) == tens