From f6a846a864112cd0df686138fae89c727e8771ab Mon Sep 17 00:00:00 2001
From: Douglas Rumbaugh <dbr4@psu.edu>
Date: Sun, 23 Jul 2023 15:38:04 -0400
Subject: BTree-based baselines for IRS and Range queries

---
 benchmarks/include/bench.h | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'benchmarks/include/bench.h')

diff --git a/benchmarks/include/bench.h b/benchmarks/include/bench.h
index e0f4c1d..28c1e97 100644
--- a/benchmarks/include/bench.h
+++ b/benchmarks/include/bench.h
@@ -49,7 +49,11 @@ static bool insert_tput_bench(DE &de_index, std::fstream &file, size_t insert_cn
         for (size_t i=0; i<insert_vec.size(); i++) {
             // process a delete if necessary
             if (applied_deletes < delete_cnt && delete_idx < delete_vec.size() && gsl_rng_uniform(g_rng) < delete_prop) {
-                de_index.erase(delete_vec[delete_idx++]);
+                if constexpr (std::is_same_v<TreeMap, DE>) {
+                    de_index.erase_one(delete_vec[delete_idx++].key);
+                } else {
+                    de_index.erase(delete_vec[delete_idx++]);
+                }
                 applied_deletes++;
             }
 
-- 
cgit v1.2.3