Machine Learning: Welcome to the matrix

Question

I am trying to solve "Welcome to the matrix" in the Machine Learning course.
Test Case 5 is showing incorrect output, rest have passed successfully. Why is it so?

Code:

https://code.sololearn.com/cZLS3t0mVQKo/?ref=app

Answer 1

Jan Markus Oh yeah 😬 I didn't realise this. Thanks for help!

Edit: It worked

Answer 2

Jan Markus any help would be appreciated