diff --git a/keras_tuner/engine/oracle.py b/keras_tuner/engine/oracle.py index 6198a3c24..563ff00f5 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