Try to fix some warnings

optimagic-dev · Dec 1, 2024 · fe27a3b · fe27a3b
1 parent e72608b
commit fe27a3b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/optimagic/benchmarking/run_benchmark.py b/src/optimagic/benchmarking/run_benchmark.py
@@ -209,7 +209,7 @@ def _process_one_result(optimize_result, problem):
             criterion_history = history.fun
         criterion_history = np.clip(criterion_history, _solution_crit, np.inf)
         batches_history = history.batches
-        time_history = history.time
+        time_history = history.start_time
 
     return {
         "params_history": params_history_flat,

diff --git a/src/optimagic/optimization/history.py b/src/optimagic/optimization/history.py
@@ -251,7 +251,7 @@ def task(self) -> list[EvalTask]:
     def time(self) -> list[float]:
         msg = (
             "The attribute `time` of History will be deprecated soon. Use the "
-            f"`{self._get_time.__name__}` method instead."
+            "`start_time` method instead."
         )
         warnings.warn(msg, FutureWarning)
         arr = np.array(self._start_time)
@@ -267,7 +267,7 @@ def criterion(self) -> list[float | None]:
     def runtime(self) -> list[float]:
         msg = (
             "The attribute `runtime` of History will be deprecated soon. Use the "
-            f"`{self._get_time.__name__}` method instead."
+            "`start_time` method instead."
         )
         warnings.warn(msg, FutureWarning)
         return self.time