KeyError in trial.py from release 1.2.0 on

[afreireosorio]

Hello!

I noticed that my code stopped working since I upgraded keras-tuner. The two pieces of code that started showing error messages are the following. The errors doesn't occur in release 1.1.3.

Thanks!

1)
Code
from kerastuner_tensorboard_logger import ( TensorBoardLogger, setup_tb # Optional )

Error msg:
generate_model.py", line 7, in
from fit_base_model import FitBaseModel
File "/home/nonroot/federated_learning/generate_model/fit_base_model.py", line 23, in
from kerastuner_tensorboard_logger import (
File "/usr/local/lib/python3.8/dist-packages/kerastuner_tensorboard_logger/init.py", line 1, in
from .logger import TensorBoardLogger
File "/usr/local/lib/python3.8/dist-packages/kerastuner_tensorboard_logger/logger.py", line 5, in
from kerastuner.engine.logger import Logger
File "/usr/local/lib/python3.8/dist-packages/kerastuner/engine/logger.py", line 1, in
from keras_tuner.engine.logger import *
ModuleNotFoundError: No module named 'keras_tuner.engine.logger'

2)
Code
tuner = kt.BayesianOptimization(hypermodel=self.hyper_model_build, objective=kt.Objective('val_avg_multilabel_BA_2', direction="max"), max_trials=15, executions_per_trial=1, overwrite=False, directory=f'{curr_dir}/finetunning', project_name='BayesianOptimization')

Error msg:
File "/usr/local/lib/python3.8/dist-packages/keras_tuner/tuners/bayesian.py", line 387, in init
super().init(oracle=oracle, hypermodel=hypermodel, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/keras_tuner/engine/tuner.py", line 113, in init
super().init(
File "/usr/local/lib/python3.8/dist-packages/keras_tuner/engine/base_tuner.py", line 117, in init
self.reload()
File "/usr/local/lib/python3.8/dist-packages/keras_tuner/engine/base_tuner.py", line 441, in reload
self.oracle.reload()
File "/usr/local/lib/python3.8/dist-packages/keras_tuner/engine/oracle.py", line 579, in reload
trial = trial_module.Trial.from_state(trial_state)
File "/usr/local/lib/python3.8/dist-packages/keras_tuner/engine/trial.py", line 165, in from_state
trial.set_state(state)
File "/usr/local/lib/python3.8/dist-packages/keras_tuner/engine/trial.py", line 160, in set_state
self.message = state["message"]
KeyError: 'message'

[haifeng-jin]

1. The kerastuner_tensorboard_logger is no longer needed to use TensorBoard.
   You can refer to: https://keras.io/guides/keras_tuner/visualize_tuning/

2. The error message looks like you are loading a 1.1.x saved directory with 1.2.x, which are not compatible. If you try with overwrite=True, it should work.

[afreireosorio]

Thanks for your precise answer, @haifeng-jin