From 347f4b2136a6cba76550645c788779a05f6847df Mon Sep 17 00:00:00 2001 From: Varun Agrawal Date: Thu, 15 May 2025 18:08:26 -0400 Subject: [PATCH] update wrapper --- gtsam/discrete/discrete.i | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/gtsam/discrete/discrete.i b/gtsam/discrete/discrete.i index 27f1fdfa1..2f6687d44 100644 --- a/gtsam/discrete/discrete.i +++ b/gtsam/discrete/discrete.i @@ -138,10 +138,13 @@ virtual class DiscreteConditional : gtsam::DecisionTreeFactor { gtsam::DecisionTreeFactor* likelihood( const gtsam::DiscreteValues& frontalValues) const; gtsam::DecisionTreeFactor* likelihood(size_t value) const; - size_t sample(const gtsam::DiscreteValues& parentsValues) const; - size_t sample(size_t value) const; - size_t sample() const; - void sampleInPlace(gtsam::DiscreteValues @parentsValues) const; + size_t sample(const gtsam::DiscreteValues& parentsValues, + std::mt19937_64 @rng = &kRandomNumberGenerator) const; + size_t sample(size_t value, + std::mt19937_64 @rng = &kRandomNumberGenerator) const; + size_t sample(std::mt19937_64 @rng = &kRandomNumberGenerator) const; + void sampleInPlace(gtsam::DiscreteValues @parentsValues, + std::mt19937_64 @rng = &kRandomNumberGenerator) const; size_t argmax(const gtsam::DiscreteValues& parentsValues) const; // Markdown and HTML @@ -233,8 +236,11 @@ class DiscreteBayesNet { double evaluate(const gtsam::DiscreteValues& values) const; double operator()(const gtsam::DiscreteValues& values) const; - gtsam::DiscreteValues sample() const; - gtsam::DiscreteValues sample(gtsam::DiscreteValues given) const; + gtsam::DiscreteValues sample(std::mt19937_64 + @rng = &kRandomNumberGenerator) const; + gtsam::DiscreteValues sample(gtsam::DiscreteValues given, + std::mt19937_64 + @rng = &kRandomNumberGenerator) const; string dot( const gtsam::KeyFormatter& keyFormatter = gtsam::DefaultKeyFormatter,