Hi @Wz111 , thanks for your question. I assume this is the assignment for week 1 of the DLS course 1 and not DLS course 2?
If that is the case, you are at exercise 8. Have a look at what is written down at exercise 7 - predict. The arguments of the predict function show the different input parameters as well as their size and data type. Does that help you out to figure out what went wrong with w?
Just adding on top of @sjfischer suggestions, the error you are getting is not related with the shape of w but rather with the contents of w, you can see that the failing assert is that one, the shape is checked in the previous step and there is no error.