From 160f44bf2588084497ed2012731b4924ce7b37ef Mon Sep 17 00:00:00 2001 From: "tor.erlend95@gmail.com" Date: Sat, 12 Sep 2020 14:44:26 +0200 Subject: [PATCH] added a test for RQS to interface.jl but should move away from this --- test/interface.jl | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/interface.jl b/test/interface.jl index f87b2bca..554066d3 100644 --- a/test/interface.jl +++ b/test/interface.jl @@ -7,7 +7,7 @@ using Tracker using DistributionsAD using Bijectors -using Bijectors: Log, Exp, Shift, Scale, Logit, SimplexBijector, PDBijector, Permute, PlanarLayer, RadialLayer, Stacked, TruncatedBijector, ADBijector +using Bijectors: Log, Exp, Shift, Scale, Logit, SimplexBijector, PDBijector, Permute, PlanarLayer, RadialLayer, Stacked, TruncatedBijector, ADBijector, RationalQuadraticSpline Random.seed!(123) @@ -159,7 +159,8 @@ end (SimplexBijector(), mapslices(z -> normalize(z, 1), rand(2, 3); dims = 1)), (stack(Exp{0}(), Scale(2.0)), randn(2, 3)), (Stacked((Exp{1}(), SimplexBijector()), [1:1, 2:3]), - mapslices(z -> normalize(z, 1), rand(3, 2); dims = 1)) + mapslices(z -> normalize(z, 1), rand(3, 2); dims = 1)), + (RationalQuadraticSpline(randn(3), randn(3), randn(3 - 1), 2.), [-0.5, 0.5]) ] for (b, xs) in bs_xs