Separation of data preparation and plotting for criterion_plot()

src/optimagic/optimization/optimize_result.py

@@ -4,6 +4,7 @@
 
 import numpy as np
 import pandas as pd
+from numpy.typing import NDArray
 
 from optimagic import deprecations
 from optimagic.logging.logger import LogReader
@@ -210,15 +211,15 @@ class MultistartInfo:
         start_parameters: List of start parameters for each optimization.
         local_optima: List of optimization results.
         exploration_sample: List of parameters used for exploration.
-        exploration_results: List of function values corresponding to exploration.
+        exploration_results: Array of function values corresponding to exploration.
         n_optimizations: Number of local optimizations that were run.
 
     """
 
     start_parameters: list[PyTree]
     local_optima: list[OptimizeResult]
     exploration_sample: list[PyTree]
-    exploration_results: list[float]
+    exploration_results: NDArray[np.float64]
 
     def __getitem__(self, key):
         deprecations.throw_dict_access_future_warning(key, obj_name=type(self).__name__)