From 2dc38f8c5eff714bfc8a385cdfa8a2da5e078f99 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Sun, 1 Sep 2024 15:47:43 -0400 Subject: [PATCH] chore: run formatter --- Project.toml | 2 +- test/layers_tests.jl | 10 ++++------ 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/Project.toml b/Project.toml index 6ea52754..f9aca992 100644 --- a/Project.toml +++ b/Project.toml @@ -58,7 +58,7 @@ SciMLBase = "2" SciMLSensitivity = "7.43" StableRNGs = "1.0.2" Statistics = "1.10" -SteadyStateDiffEq = "2" +SteadyStateDiffEq = "2.3.2" Test = "1.10" Zygote = "0.6.69" julia = "1.10" diff --git a/test/layers_tests.jl b/test/layers_tests.jl index 62d10592..c247d9cb 100644 --- a/test/layers_tests.jl +++ b/test/layers_tests.jl @@ -55,8 +55,7 @@ end x = randn(rng, Float32, x_size...) |> dev z, st = model(x, ps, st) - opt_broken = solver isa SimpleLimitedMemoryBroyden - @jet model(x, ps, st) opt_broken=opt_broken + @jet model(x, ps, st) opt_broken=true @test all(isfinite, z) @test size(z) == size(x) @@ -102,8 +101,8 @@ end dense_layer(2 => 4) dense_layer(2 => 3) NoOpLayer() dense_layer(2 => 1); dense_layer(1 => 4) dense_layer(1 => 3) dense_layer(1 => 2) NoOpLayer()]] - init_layers = [(dense_layer(4 => 4), dense_layer(4 => 3), - dense_layer(4 => 2), dense_layer(4 => 1))] + init_layers = [( + dense_layer(4 => 4), dense_layer(4 => 3), dense_layer(4 => 2), dense_layer(4 => 1))] x_sizes = [(4, 3)] scales = [((4,), (3,), (2,), (1,))] @@ -141,8 +140,7 @@ end z, st = model(x, ps, st) z_ = DEQs.__flatten_vcat(z) - opt_broken = solver isa SimpleLimitedMemoryBroyden - @jet model(x, ps, st) opt_broken=opt_broken # Broken due to nfe dynamic dispatch + @jet model(x, ps, st) opt_broken=true @test all(isfinite, z_) @test size(z_) == (sum(prod, scale), size(x, ndims(x)))