Course 2 Week 1 Assignment Initialization Training Error

Hello, my code passed all tests, but I am getting an error while training the model. The function looks correct. I did not write this code snippet and I did not change it. Could you please help me?

Double-check how you are implementing the initialize_parameters_he function.