Data generator exercise

I’m getting some tests passed, and some failed for the batch generator question in week2 for course 3. Would appreciate any help and my labid is: escfgbtrrtec

This is the error I get:
Element with index 2 in the output tuple has incorrect type.
Expected <class ‘jaxlib.xla_extension.DeviceArrayBase’>.
Got <class ‘numpy.ndarray’>.
Element with index 2 in the output tuple has incorrect type.
Expected <class ‘jaxlib.xla_extension.DeviceArrayBase’>.
Got <class ‘numpy.ndarray’>.
Element with index 2 in the output tuple has incorrect shape. It should be (batch_size, max_length).
Expected (4, 11).
Got (4,).
Element with index 2 in the output tuple has incorrect shape. It should be (batch_size, max_length).
Expected [[1 1 1 1 1 1 1 1 0 0 0]
[1 1 1 1 1 1 1 1 1 1 0]
[1 1 1 1 1 1 1 1 1 1 0]
[1 1 1 1 1 1 1 1 0 0 0]].
Got [list([1, 1, 1, 1, 1, 1, 1, 1]) list([1, 1, 1, 1, 1, 1, 1, 1, 1, 1])
list([1, 1, 1, 1, 1, 1, 1, 1, 1, 1]) list([1, 1, 1, 1, 1, 1, 1, 1])].
Element with index 2 in the output tuple has incorrect type.
Expected <class ‘jaxlib.xla_extension.DeviceArrayBase’>.
Got <class ‘numpy.ndarray’>.
25 Tests passed
5 Tests failed

please ignore the question, I was able to figure out the error. Thanks, Rama