M4ML Course 1 week 3 assignment

Exercise #3:
My code produces and matches expected output but the test reports an error:

Not clear on the Assert trap nor how to edit code without getting incorrect values.
I’ve reloaded, restarted kernel and still no change - output matches expected output but test still fails.

AssertionError Traceback (most recent call last)
in
1 # Note:
2 # Actual values are not checked here in the unit tests (due to random initialization).
----> 3 w3_unittest.test_initialize_parameters(initialize_parameters)

~/work/w3_unittest.py in test_initialize_parameters(target_initialize_parameters)
185
186 for test_case in test_cases:
→ 187 result = target_initialize_parameters(test_case[“input”][“n_x”], test_case[“input”][“n_y”])
188
189 try:

in initialize_parameters(n_x, n_y)
14 ### END CODE HERE ###
15
—> 16 assert (W.shape == (n_y, n_x))
17 assert (b.shape == (n_y, 1))
18

AssertionError:

probably because:
W = np.random.randn(n_x,n_y) * 0.01
instead of
W = np.random.randn(n_y,n_x) * 0.01