Merge pull request #78 from iraikov/fix/0-stats

Prevent error for 0-length stats edge case
iraikov · Apr 13, 2024 · 520cc61 · 520cc61
2 parents f27763f + ccb73fe
commit 520cc61
Showing 1 changed file with 22 additions and 10 deletions.
diff --git a/dmosopt/dmosopt.py b/dmosopt/dmosopt.py
@@ -270,16 +270,28 @@ def _update_evals(self):
             else:
                 self.t = np.vstack((self.t, t_completed))
             ts = self.t[self.t > 0.0]
-            self.stats.update(
-                {
-                    "eval_min": np.min(ts),
-                    "eval_max": np.max(ts),
-                    "eval_mean": np.mean(ts),
-                    "eval_std": np.std(ts),
-                    "eval_sum": np.sum(ts),
-                    "eval_median": np.median(ts),
-                }
-            )
+            if len(ts) > 0:
+                self.stats.update(
+                    {
+                        "eval_min": np.min(ts),
+                        "eval_max": np.max(ts),
+                        "eval_mean": np.mean(ts),
+                        "eval_std": np.std(ts),
+                        "eval_sum": np.sum(ts),
+                        "eval_median": np.median(ts),
+                    }
+                )
+            else:
+                self.stats.update(
+                    {
+                        "eval_min": -1,
+                        "eval_max": -1,
+                        "eval_mean": -1,
+                        "eval_std": -1,
+                        "eval_sum": -1,
+                        "eval_median": -1,
+                    }
+                )
 
             self._remove_duplicate_evals()
             self.completed = []