Update keras/src/layers/layer_test.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: divyashreepathihalli

keras/src/layers/layer_test.py

@@ -1765,6 +1765,7 @@ def test_quantized_model_save_and_load(self):
         inputs = layers.Input(shape=(None,))
         x = layers.Embedding(input_dim=10, output_dim=10)(inputs)
         x = layers.Dense(10)(x)
+        x = layers.BatchNormalization()(x) # Add batchnorm
         model = models.Model(inputs=inputs, outputs=x)
         path = os.path.join(self.get_temp_dir(), "quantized_model.keras")
         model.quantize(mode="int8")