Fixed incorrect reference in DistributionsAD extension (#299)

* fixed incorrect reference in DistributionsAD extension * bump patch version * added tests for #298 * bump patch version
TuringLang · Apr 19, 2024 · 0839209 · 0839209 · torfjelde · Apr 19, 2024
1 parent e522155
commit 0839209
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 2 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.10"
 
 [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/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"