fixed test_utils

TuringLang · Feb 3, 2023 · 8d77d78 · 8d77d78
1 parent dfaf7be
commit 8d77d78
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/test_utils.jl b/src/test_utils.jl
@@ -686,7 +686,7 @@ Simple model for which [`default_transformation`](@ref) returns a [`StaticTransf
 end
 
 function DynamicPPL.default_transformation(::Model{typeof(demo_static_transformation)})
-    b = Bijectors.stack(exp, identity)
+    b = Bijectors.stack(Bijectors.elementwise(exp), identity)
     return DynamicPPL.StaticTransformation(b)
 end