C3W2_Assignment Week 2: Diving deeper into the BBC News archive

Hello, I have an issue in preprocess_dataset and I do not know exactly how to solve.

The shape is as expected, however when I run the unittest I receive an error.
What is wrong ?

The 1st expected output section looks only at the shape of X (note the train_batch[0]). The unit test checks the shapes of both the model inputs and labels.

Your implementation is supposed to return a dataset made of tuples. Each tuple should contain model inputs at the 0th index and the corresponding labels at index 1.