fix bug in univariate sampling

Red-Portal · Aug 11, 2024 · 4a245a5 · 4a245a5
1 parent ee338f5
commit 4a245a5
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/src/univariate.jl b/src/univariate.jl
@@ -95,6 +95,14 @@ function AbstractMCMC.step(rng    ::Random.AbstractRNG,
     return t, UnivariateSliceState(t)
 end
 
+struct UnivariateTarget{Model}
+    model::Model
+end
+
+function LogDensityProblems.logdensity(uni::UnivariateTarget, θi)
+    LogDensityProblems.logdensity(uni.model, [θi])
+end
+
 function AbstractMCMC.step(
     rng    ::Random.AbstractRNG,
     model  ::AbstractMCMC.LogDensityModel, 

diff --git a/test/multivariate.jl b/test/multivariate.jl
@@ -57,7 +57,7 @@ function LogDensityProblems.dimension(model::MultiModel)
     2
 end
 
-@testset "sampling" begin
+@testset "multivariate sampling" begin
     model   = MultiModel(1., 1., [0.])
     @testset for sampler in [
         # Vector-valued windows

diff --git a/test/univariate.jl b/test/univariate.jl
@@ -45,7 +45,7 @@ function LogDensityProblems.dimension(model::UniModel)
     1
 end
 
-@testset "sampling" begin
+@testset "multivariate sampling" begin
     model   = UniModel([0.])
     @testset for sampler in [
         Slice(1),