Fix stats dtype

icaros-usc · Jun 14, 2024 · 2662980 · 2662980
1 parent 92f0e7b
commit 2662980
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ribs/archives/_archive_base.py b/ribs/archives/_archive_base.py
@@ -335,9 +335,11 @@ def _stats_update(self, new_objective_sum, new_best_index):
             coverage=np_scalar(len(self) / self.cells,
                                dtype=self.dtypes["objective"]),
             qd_score=new_qd_score,
-            norm_qd_score=new_qd_score / self.cells,
+            norm_qd_score=np_scalar(new_qd_score / self.cells,
+                                    dtype=self.dtypes["objective"]),
             obj_max=new_obj_max,
-            obj_mean=self._objective_sum / len(self),
+            obj_mean=np_scalar(self._objective_sum / len(self),
+                               dtype=self.dtypes["objective"]),
         )
 
     def add(self, solution, objective, measures, **fields):