chore: Update Display class in oracle.py with col_width parameter

Fixes keras-team#1024
Anselmoo · Aug 30, 2024 · 3a0f6ce · 3a0f6ce
1 parent 417e5b5
commit 3a0f6ce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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