Convolutional_block week 2 assignment_1

Thanks, It work but now showing this error although the expected value is correct.
X_shortcut = BatchNormalization(axis = 3)(X_shortcut)
error
tf.Tensor(
[[[0. 0.66683817 0. 0. 0.88853896 0.5274254 ]
[0. 0.65053666 0. 0. 0.89592844 0.49965227]]

[[0. 0.6312079 0. 0. 0.8636247 0.47643146]
[0. 0.5688321 0. 0. 0.85534114 0.41709304]]], shape=(2, 2, 6), dtype=float32)

AssertionError Traceback (most recent call last)
in
16
17 B = convolutional_block(X, f = 2, filters = [2, 4, 6], training=True)
—> 18 assert np.allclose(B.numpy(), convolutional_block_output2), “Wrong values when training=True.”
19
20 print(’\033[92mAll tests passed!’)

AssertionError: Wrong values when training=True.

Expected value

tf.Tensor(
[[[0. 0.66683817 0. 0. 0.88853896 0.5274254 ]
[0. 0.65053666 0. 0. 0.89592844 0.49965227]]

[[0. 0.6312079 0. 0. 0.8636247 0.47643146]
[0. 0.5688321 0. 0. 0.85534114 0.41709304]]], shape=(2, 2, 6), dtype=float32)