Revert threshold adding

icaros-usc · Nov 8, 2023 · e5f2e81 · e5f2e81
1 parent 9713f6f
commit e5f2e81
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ribs/archives/_archive_base.py b/ribs/archives/_archive_base.py
@@ -275,6 +275,7 @@ def _stats_update(self, new_objective_sum, new_best_index):
                 new_best_elite["objective"] > self._stats.obj_max):
             # Convert batched values to single values.
             new_best_elite = {k: v[0] for k, v in new_best_elite.items()}
+            new_best_elite.pop("threshold")
 
             new_obj_max = new_best_elite["objective"]
             self._best_elite = new_best_elite