Week3 Assignment 1 - yolo_filter_boxes

What is supposed to be the dimension of box_class_scores ?
I am applying the reduce_max with axis=-1 on box_scores to get the max class value of each box.
but it leads to an error : AssertionError: Wrong shape in scores later on.

I added print statements to my code to show the shape and type of everything. Here’s what I see when I run that test cell for yolo_filter_boxes:

boxes.shape (19, 19, 5, 4)
boxes.dtype <dtype: 'float32'>
box_scores.shape (19, 19, 5, 80)
box_scores.dtype <dtype: 'float32'>
box_classes.shape (19, 19, 5)
box_classes.dtype <dtype: 'int64'>
box_class_scores.shape (19, 19, 5)
box_class_scores.dtype <dtype: 'float32'>
filtering_mask.shape (19, 19, 5)
filtering_mask.dtype <dtype: 'bool'>
scores[2] = 9.270486
boxes[2] = [ 4.6399336  3.2303846  4.431282  -2.202031 ]
classes[2] = 8
scores.shape = (1789,)
boxes.shape = (1789, 4)
classes.shape = (1789,)
 All tests passed!

Thank you. I have found my bug.