Assertion error in Exercise 2

I have tried to code initialize_parameters functions but I keep ending up with assertion errors. I have coded the above function correctly i.e., layer_sizes

Issue was from my end. I had accidentally modified the test case.