Quick-fix for #100 (#101)

* fix for #100 * fix link and invlink Co-authored-by: mohamed82008 <[email protected]>
TuringLang · Apr 7, 2020 · 77cbd05 · 77cbd05
1 parent 2746ff9
commit 77cbd05
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/src/compat/distributionsad.jl b/src/compat/distributionsad.jl
@@ -254,11 +254,28 @@ function link(
     dist::VectorOfUnivariate,
     x::AbstractMatrix{<:Real},
 )
-    return mapvcat(eachcol(x)) do x
+    return eachcolmaphcat(x) do x
         link(dist, x)
     end
 end
 
+function invlink(
+    dist::VectorOfUnivariate,
+    x::AbstractVector{<:Real},
+)
+    return mapvcat(dist.v, x) do dist, x
+        invlink(dist, x)
+    end
+end
+function invlink(
+    dist::VectorOfUnivariate,
+    x::AbstractMatrix{<:Real},
+)
+    return eachcolmaphcat(x) do x
+        invlink(dist, x)
+    end
+end
+
 function logpdf_with_trans(
     dist::MatrixOfUnivariate,
     x::AbstractMatrix{<:Real},