Remove check-fail that priorities are >= 0.

#135 PiperOrigin-RevId: 574426193 Change-Id: I61ee2677b92365c3fb972ea39e8542a9a964ed98
google-deepmind · Oct 18, 2023 · bfa91b5 · bfa91b5
1 parent 7e9ee1a
commit bfa91b5
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/reverb/cc/selectors/prioritized.cc b/reverb/cc/selectors/prioritized.cc
@@ -17,8 +17,6 @@
 #include <cmath>
 #include <cstddef>
 
-#include "absl/random/distributions.h"
-#include "absl/random/random.h"
 #include "absl/status/status.h"
 #include "absl/strings/str_cat.h"
 #include "reverb/cc/platform/logging.h"
@@ -56,9 +54,7 @@ PrioritizedSelector::PrioritizedSelector(double priority_exponent,
     : priority_exponent_(priority_exponent),
       capacity_(std::pow(2, 17)),
       sum_tree_(capacity_),
-      rng_(seed) {
-  REVERB_CHECK_GE(priority_exponent_, 0);
-}
+      rng_(seed) {}
 
 absl::Status PrioritizedSelector::Delete(Key key) {
   const size_t last_index = key_to_index_.size() - 1;