Merge branch 'master' into torfjelde/product-bijector

TuringLang · May 7, 2024 · 9887270 · 9887270
2 parents e8fc8cf + d364639
commit 9887270
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 5 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "Bijectors"
 uuid = "76274a88-744f-5084-9051-94815aaf08c4"
-version = "0.13.9"
+version = "0.13.11"
 
 [deps]
 ArgCheck = "dce04be8-c92d-5529-be00-80e4d2c0e197"

diff --git a/ext/BijectorsDistributionsADExt.jl b/ext/BijectorsDistributionsADExt.jl
@@ -51,7 +51,7 @@ function Bijectors.bijector(d::FillMatrixOfUnivariate{Continuous})
 end
 Bijectors.bijector(d::MatrixOfUnivariate{Discrete}) = identity
 function Bijectors.bijector(d::MatrixOfUnivariate{Continuous})
-    return TruncatedBijectors.Bijector(_minmax(d.dists)...)
+    return Bijectors.TruncatedBijector(Bijectors._minmax(d.dists)...)
 end
 Bijectors.bijector(d::VectorOfMultivariate{Discrete}) = identity
 for T in (:VectorOfMultivariate, :FillVectorOfMultivariate)

diff --git a/src/bijectors/logit.jl b/src/bijectors/logit.jl
@@ -1,9 +1,9 @@
 ######################
 # Logit and Logistic #
 ######################
-struct Logit{T} <: Bijector
-    a::T
-    b::T
+struct Logit{T1,T2} <: Bijector
+    a::T1
+    b::T2
 end
 
 Functors.@functor Logit

diff --git a/test/distributionsad.jl b/test/distributionsad.jl
@@ -0,0 +1,7 @@
+@testset "DistributionsAD" begin
+    # https://github.com/TuringLang/Bijectors.jl/issues/298
+    @testset "#298" begin
+        dists = arraydist(fill(InverseGamma(), 2, 2))
+        @test bijector(dists) isa Bijectors.TruncatedBijector
+    end
+end
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -48,6 +48,8 @@ if GROUP == "All" || GROUP == "Interface"
     include("bijectors/pd.jl")
     include("bijectors/reshape.jl")
     include("bijectors/corr.jl")
+
+    include("distributionsad.jl")
 end
 
 if GROUP == "All" || GROUP == "AD"