fix

impICNF · Jun 7, 2024 · 9a07cd3 · 9a07cd3
1 parent 1d31992
commit 9a07cd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,