No module named 'keras.src.engine.base_layer_v1'

I encountered an error with this code

      6 input_shape = (X_train.shape[1],)
      8 model = Sequential([
----> 9     Dense(256, activation='relu', input_shape=input_shape),
     10     Dropout(rate=0.2),

Keras version is 2.14.0

how do I fix this
thank you

Hi @Tharindu_Madhusanka, I was able to define the dense layer with Tensorflow and keras 2.14 in colab and did not face any error. Could you please let us know in which environment you are facing this error

Thank You.