Skip to content

Commit 2e8f948

Browse files
committed
fix regressions
1 parent f853b15 commit 2e8f948

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

gtsam/discrete/tests/testDiscreteBayesNet.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -99,9 +99,9 @@ TEST(DiscreteBayesNet, Asia) {
9999

100100
// now sample from it
101101
DiscreteValues expectedSample{{Asia.first, 1}, {Dyspnea.first, 1},
102-
{XRay.first, 1}, {Tuberculosis.first, 0},
103-
{Smoking.first, 1}, {Either.first, 1},
104-
{LungCancer.first, 1}, {Bronchitis.first, 0}};
102+
{XRay.first, 0}, {Tuberculosis.first, 0},
103+
{Smoking.first, 1}, {Either.first, 0},
104+
{LungCancer.first, 0}, {Bronchitis.first, 1}};
105105
SETDEBUG("DiscreteConditional::sample", false);
106106
auto actualSample = chordal2->sample();
107107
EXPECT(assert_equal(expectedSample, actualSample));

gtsam/hybrid/HybridBayesNet.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ HybridValues HybridBayesNet::sample(const HybridValues &given,
188188
}
189189
}
190190
// Sample a discrete assignment.
191-
const DiscreteValues assignment = dbn.sample(given.discrete());
191+
const DiscreteValues assignment = dbn.sample(given.discrete(), rng);
192192
// Select the continuous Bayes net corresponding to the assignment.
193193
GaussianBayesNet gbn = choose(assignment);
194194
// Sample from the Gaussian Bayes net.

gtsam/hybrid/tests/testHybridBayesNet.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ TEST(HybridBayesNet, EvaluatePureDiscrete) {
9090

9191
// sample
9292
std::mt19937_64 rng(42);
93-
EXPECT(assert_equal(zero, bayesNet.sample(&rng)));
93+
EXPECT(assert_equal(one, bayesNet.sample(&rng)));
9494
EXPECT(assert_equal(one, bayesNet.sample(one, &rng)));
9595
EXPECT(assert_equal(zero, bayesNet.sample(zero, &rng)));
9696

@@ -616,13 +616,13 @@ TEST(HybridBayesNet, Sampling) {
616616
double discrete_sum =
617617
std::accumulate(discrete_samples.begin(), discrete_samples.end(),
618618
decltype(discrete_samples)::value_type(0));
619-
EXPECT_DOUBLES_EQUAL(0.477, discrete_sum / num_samples, 1e-9);
619+
EXPECT_DOUBLES_EQUAL(0.519, discrete_sum / num_samples, 1e-9);
620620

621621
VectorValues expected;
622622
// regression for specific RNG seed
623623
#if __APPLE__ || _WIN32
624-
expected.insert({X(0), Vector1(-0.0131207162712)});
625-
expected.insert({X(1), Vector1(-0.499026377568)});
624+
expected.insert({X(0), Vector1(0.0252479903896)});
625+
expected.insert({X(1), Vector1(-0.513637101911)});
626626
#elif __linux__
627627
expected.insert({X(0), Vector1(-0.00799425182219)});
628628
expected.insert({X(1), Vector1(-0.526463854268)});

0 commit comments

Comments
 (0)