Unable to fix the issue - C1_W4 programming assignment

Having the following issue when submitting the code for the assignment. Can you please help me with that

The problem is that you are using hard-coded values here:

These values will not work for the general case. The grader tests your code using different sizes and shapes of data.

So you need to use the shape of Y here, not any hard-coded values.