Fix: std::remove_if called without erase

intel · Dec 2, 2023 · 383dc08 · 383dc08
1 parent c6507fd
commit 383dc08
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/gen_policy.cpp b/src/gen_policy.cpp
@@ -537,7 +537,8 @@ void GenPolicy::uniformProbFromMax(std::vector<Probability<T>> &distr,
 
 template <class T, class U>
 void GenPolicy::removeProbability(std::vector<Probability<T>> &orig, U id) {
-    std::remove_if(
+    auto new_end = std::remove_if(
         orig.begin(), orig.end(),
         [&id](Probability<T> &elem) -> bool { return elem.getId() == id; });
+    orig.erase(new_end, orig.end());
 }