Week 2 Assignment 2 Exercise 2

In exercise 2 at line 30 x = data_augmentation(inputs) it shows error “Input 0 of layer sequential_2 is incompatible with the layer: expected ndim=4, found ndim=3. Full shape received: [None, 160, 160]”

It looks like you have done something that causes the “channels” dimension to get stripped from the inputs. So how could that happen?