Course 4, Week 2 Assignment 2 Problem while running and grading

I am trying to run this section of code in notebook

data_augmentation = data_augmenter()

for image, _ in train_dataset.take(1):
plt.figure(figsize=(10, 10))
first_image = image[0]
for i in range(9):
ax = plt.subplot(3, 3, i + 1)
augmented_image = data_augmentation(tf.expand_dims(first_image, 0))
plt.imshow(augmented_image[0] / 255)

and it is resulting in Dead Kernel

Also after submitting grader is failing giving this error message please help
There was an error grading one or more parts of your submission. Please review the messages below for suggestions.


Filename: ex-w2a2

0/100Score: 0 of 100

Grading failed

We encountered the following warnings when grading this part:

  • Grader error: Malformed feedback. Please visit the Discussion forum to see if your peers are experiencing similar errors. If the error isn’t resolved in 24 hours, please reach out to Coursera through our Help Center.

Perhaps there is an error in your data_augmenter() function.

But that piece of code was already present means I have not written that piece of code, also I don’t remember editing that code which was already present.

No, you have to add code to the data_augmenter() function.

Have you found a solution? I have the same problem.