Practice Lab: Exercise 3 error

Hello,

I am getting an assertion error on Exercise 3.
All the outputs for the practice lab example on Exercise 3 are matching with the results. However, I am getting this error for the test function:
Wrong information gain. Expected 0.311278 got: 0.44902249956730633

Could you please help me figure out the source of the error?

Thank you,

Problem resolved by learner.