format parameter values correctly

Author: adrianmolzon

bayes_opt/logger.py

@@ -139,8 +139,8 @@ def _format_str(self, str_: str) -> str:
 
     def _print_step(
         self,
-        result: dict[str, Any],
         keys: list[str],
+        result: dict[str, Any],
         params_config: Mapping[str, ParameterConfig],
         colour: str = _colour_regular_message,
     ) -> str:
@@ -173,9 +173,11 @@ def _print_step(
             cells[2] = self._format_bool(result["allowed"])
         params = result.get("params", {})
         cells[3:] = [
-            params_config[key].to_string(val, self._default_cell_size) for key, val in params.items()
+            self._format_number(val)
+            if isinstance(val, (int, float))
+            else params_config[key].to_string(val, self._default_cell_size)
+            for key, val in params.items()
         ]
-
         return "| " + " | ".join(colour + x + self._colour_reset for x in cells if x is not None) + " |"
 
     def _print_header(self, keys: list[str]) -> str:
@@ -282,7 +284,7 @@ def log_optimization_step(
 
         if self._verbose == 2 or is_new_max:
             colour = self._colour_new_max if is_new_max else self._colour_regular_message
-            line = self._print_step(result, keys, params_config, colour=colour) + "\n"
+            line = self._print_step(keys, result, params_config, colour=colour) + "\n"
             if self._verbose:
                 print(line, end="")
 

tests/test_logger.py

@@ -148,7 +148,7 @@ def test_step():
 
     # Test default color
     step_str = logger._print_step(
-        optimizer._space.res()[-1], optimizer._space.keys, optimizer._space.params_config
+        optimizer._space.keys, optimizer._space.res()[-1], optimizer._space.params_config
     )
     assert "|" in step_str
     assert "1" in step_str  # iteration
@@ -157,7 +157,7 @@ def test_step():
     # Test with custom color
     custom_color = Fore.RED
     step_str_colored = logger._print_step(
-        optimizer._space.res()[-1], optimizer._space.keys, optimizer._space.params_config, colour=custom_color
+        optimizer._space.keys, optimizer._space.res()[-1], optimizer._space.params_config, colour=custom_color
     )
     assert custom_color in step_str_colored