fix broadcasting

Author: tpapp

CHANGELOG.md

@@ -4,6 +4,8 @@
 
 - rewrite internals to work better with AD (especially Zygote)
 
+- fix broadcasting (`Ref(transformation)` no longer necessary)
+
 # 0.3.4
 
 - make `inverse(::ArrayTransform)` accept `AbstractArray`

src/generic.jl

@@ -108,6 +108,8 @@ The user interface consists of
 """
 abstract type AbstractTransform end
 
+Base.broadcastable(t::AbstractTransform) = Ref(t)
+
 """
 $(TYPEDEF)
 

test/runtests.jl

@@ -449,3 +449,16 @@ end
     t = as(Array, 2, 3)
     @test inverse(t, ones(SMatrix{2,3})) == ones(6)
 end
+
+####
+#### broadcasting
+####
+
+@testset "broadcasting" begin
+    @test as𝕀.([0, 0]) == [0.5, 0.5]
+
+    t = UnitVector(3)
+    d = dimension(t)
+    x = [zeros(d), zeros(d)]
+    @test t.(x) == map(t, x)
+end