fix: Incorrectly modified ScreenLogger

bayes_opt/logger.py

@@ -5,6 +5,7 @@
 import json
 from contextlib import suppress
 from pathlib import Path
+from typing import Any
 
 import numpy as np
 from colorama import Fore, just_fix_windows_console
@@ -153,15 +154,16 @@ def _step(self, instance, colour=_colour_regular_message):
         -------
         A stringified, formatted version of the most recent optimization step.
         """
-        res = instance.res[-1]
-        cells: list[str | None] = [None] * 4
+        res: dict[str, Any] = instance.res[-1]
+        keys: list[str] = instance.space.keys
+        # iter, target, allowed [, *params]
+        cells: list[str | None] = [None] * (3 + len(keys))
 
         cells[:2] = self._format_number(self._iterations + 1), self._format_number(res["target"])
         if self._is_constrained:
             cells[2] = self._format_bool(res["allowed"])
-
-        for key in instance.space.keys:
-            cells[3] = self._format_number(res["params"][key])
+        params = res.get("params", {})
+        cells[3:] = [self._format_number(params.get(key, float("nan"))) for key in keys]
 
         return "| " + " | ".join(colour + x + self._colour_reset for x in cells if x is not None) + " |"
 
@@ -177,14 +179,14 @@ def _header(self, instance):
         -------
         A stringified, formatted version of the most header.
         """
-        cells: list[str | None] = [None] * 4
+        keys: list[str] = instance.space.keys
+        # iter, target, allowed [, *params]
+        cells: list[str | None] = [None] * (3 + len(keys))
 
         cells[:2] = self._format_key("iter"), self._format_key("target")
         if self._is_constrained:
             cells[2] = self._format_key("allowed")
-
-        for key in instance.space.keys:
-            cells[3] = self._format_key(key)
+        cells[3:] = [self._format_key(key) for key in keys]
 
         line = "| " + " | ".join(x for x in cells if x is not None) + " |"
         self._header_length = len(line)