From 9a07cd3f6ebd0cbe058978973140698db87e56a5 Mon Sep 17 00:00:00 2001 From: Hossein Pourbozorg Date: Fri, 7 Jun 2024 09:59:32 +0330 Subject: [PATCH] fix --- test/regression_tests.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/regression_tests.jl b/test/regression_tests.jl index 74ae9946..27ad09ec 100644 --- a/test/regression_tests.jl +++ b/test/regression_tests.jl @@ -4,7 +4,7 @@ Test.@testset "Regression Tests" begin naugs = nvars n_in = nvars + naugs n = 2^10 - nn = Chain(Dense(n_in => 3 * n_in, tanh), Dense(3 * n_in => n_in, tanh)) + nn = Lux.Chain(Lux.Dense(n_in => 3 * n_in, tanh), Lux.Dense(3 * n_in => n_in, tanh)) icnf = ContinuousNormalizingFlows.construct( ContinuousNormalizingFlows.RNODE,