diff --git a/keras_tuner/engine/oracle.py b/keras_tuner/engine/oracle.py index 6198a3c24..42f681f2f 100644 --- a/keras_tuner/engine/oracle.py +++ b/keras_tuner/engine/oracle.py @@ -116,10 +116,10 @@ def wrapped_func(*args, **kwargs): # TODO: Add more extensive display. class Display(stateful.Stateful): - def __init__(self, oracle, verbose=1): + def __init__(self, oracle, verbose=1, col_width=18): self.verbose = verbose self.oracle = oracle - self.col_width = 18 + self.col_width = col_width # Start time for the overall search self.search_start = None @@ -384,6 +384,14 @@ def verbose(self, value): value = 1 self._display.verbose = value + @property + def col_width(self): + return self._display.col_width + + @col_width.setter + def col_width(self, value): + self._display.col_width = value + def _populate_space(self, trial_id): warnings.warn( "The `_populate_space` method is deprecated, " diff --git a/keras_tuner/engine/oracle_test.py b/keras_tuner/engine/oracle_test.py index 4a584831e..8ed3e2ffe 100644 --- a/keras_tuner/engine/oracle_test.py +++ b/keras_tuner/engine/oracle_test.py @@ -450,3 +450,9 @@ def test_display_format_duration_large_d(): oracle.verbose = "auto" assert oracle_module.Display(oracle).format_duration(d) == "7d 00h 00m 00s" assert oracle.verbose == 1 + + +def test_display_col_width() -> None: + oracle = gridsearch.GridSearchOracle() + oracle.col_width = 10 + assert oracle_module.Display(oracle).col_width == 10