Update runtests.jl

YichengDWu · Mar 18, 2023 · 3b7c8d2 · 3b7c8d2
1 parent cfb8b7d
commit 3b7c8d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -189,7 +189,7 @@ rng = Random.default_rng()
             @testset "Constructors" begin
                 model = DeepONet((3, 5, 4), relu, (2, 6, 4, 4), tanh)
                 @test model.branch_net.layers[end].activation == identity
-                @test model.trunk_net.layers[end].activation == tanh_fast
+                @test model.trunk_net.layers[end].activation == tanh
 
                 branch = Chain(Dense(2, 3), Dense(3, 4))
                 trunk = Chain(Dense(3, 4), Dense(4, 5))