From 630acafbb8a0dce46b58dad740413df32e930284 Mon Sep 17 00:00:00 2001 From: Bagaev Dmitry Date: Thu, 30 Jan 2025 14:20:32 +0100 Subject: [PATCH] style: make format --- src/rules/binomial_polya/y.jl | 4 ++-- test/rules/binomial_polya/y_tests.jl | 6 ++---- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/src/rules/binomial_polya/y.jl b/src/rules/binomial_polya/y.jl index 4cf0e2e42..8619ac8ff 100644 --- a/src/rules/binomial_polya/y.jl +++ b/src/rules/binomial_polya/y.jl @@ -11,7 +11,7 @@ using ExponentialFamily.LogExpFunctions else n_samples = getn_samples(meta) βsamples = rand(meta.rng, q_β, n_samples) - p_avg = mapreduce(βsample -> logistic(dot(x, βsample)), +, eachcol(βsamples))/n_samples + p_avg = mapreduce(βsample -> logistic(dot(x, βsample)), +, eachcol(βsamples)) / n_samples return Binomial(n, p_avg) end -end \ No newline at end of file +end diff --git a/test/rules/binomial_polya/y_tests.jl b/test/rules/binomial_polya/y_tests.jl index dc8337163..b8f92784c 100644 --- a/test/rules/binomial_polya/y_tests.jl +++ b/test/rules/binomial_polya/y_tests.jl @@ -8,12 +8,12 @@ q_x = PointMass([0.1, 0.2]) q_n = PointMass(5) q_β = MvNormalWeightedMeanPrecision([3.0, -1.0], diageye(2)) - + # Test with default meta pred_dist = @call_rule BinomialPolya(:y, Marginalisation) (q_x = q_x, q_n = q_n, q_β = q_β, meta = nothing) @test pred_dist isa Binomial @test ntrials(pred_dist) == 5 - + # Test with Monte Carlo sampling meta = BinomialPolyaMeta(1000, MersenneTwister(42)) pred_dist_mc = @call_rule BinomialPolya(:y, Marginalisation) (q_x = q_x, q_n = q_n, q_β = q_β, meta = meta) @@ -23,6 +23,4 @@ @test pred_dist_mc.p ≈ pred_dist.p atol = 1e-2 end - end -