Describe why we convert to float

optimagic-dev · Mar 6, 2024 · df15ada · df15ada
1 parent bb0d0e3
commit df15ada
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/estimagic/visualization/profile_plot.py b/src/estimagic/visualization/profile_plot.py
@@ -164,8 +164,9 @@ def create_solution_times(df, runtime_measure, converged_info, return_tidy=True)
 
     """
     solution_times = df.groupby(["problem", "algorithm"])[runtime_measure].max()
-    solution_times = solution_times.unstack().astype(float)
-    solution_times = solution_times.where(converged_info, other=np.inf)
+    solution_times = solution_times.unstack()
+    # We convert the dtype to float to support the use of np.inf
+    solution_times = solution_times.astype(float).where(converged_info, other=np.inf)
 
     if not return_tidy:
         solution_times = solution_times.stack().reset_index()