Merge pull request #78 from probcomp/070924-thomaswc-fixbuild

Fix build by upgrading CategoricalEmission::incorporate to take a weight
probcomp · Jul 9, 2024 · 1276c8f · 1276c8f
2 parents 208bcf2 + cdc0e40
commit 1276c8f
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/cxx/emissions/categorical.hh b/cxx/emissions/categorical.hh
@@ -20,14 +20,9 @@ class CategoricalEmission : public Emission<int> {
     }
   };
 
-  void incorporate(const std::pair<int, int>& x) {
-    ++N;
-    emission_dists[x.first].incorporate(x.second);
-  }
-
-  void unincorporate(const std::pair<int, int>& x) {
-    --N;
-    emission_dists[x.first].unincorporate(x.second);
+  void incorporate(const std::pair<int, int>& x, double weight = 1.0) {
+    N += weight;
+    emission_dists[x.first].incorporate(x.second, weight);
   }
 
   double logp(const std::pair<int, int>& x) const {