Many errors when using Keras 2.2 version

want to replace below Dense layer with BayesianDense.

ct_input = Input(shape=(PATCH_HEIGHT, PATCH_WIDTH, 1))
pet_input = Input(shape=(PATCH_HEIGHT, PATCH_WIDTH, 1))

x = concatenate([ct_model, pet_model])
x = Dense(864, activation='relu')(x)
x = Dense(288, activation='relu')(x)
output = Dense(1, activation='sigmoid')(x)

I got many errors. E,g: ImportError: cannot import name... __Init__ and __Build__
Do you have any example to use with relevant output graphs.