Problem submitting C2_W4

Hello! I did the code and I got the expected output but when I submit the assigment I got 0% and this message:

Cell #11. Can’t compile the student’s code. Error: AssertionError(‘Wrong information gain. Expected 0.0 got: nan’)
Traceback (most recent call last):
File “/home/www/app/grading/”, line 112, in handle_solution_errors
yield {}
File “/home/www/app/grading/”, line 393, in _grade
context =
File “/home/www/app/grading/submission/”, line 195, in run
return list(self._code_items.values())[cell_num - 1].run()
File “/home/www/app/grading/submission/”, line 54, in run
return import_module(self.import_statement, items)
File “/usr/local/lib/python3.7/importlib/”, line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File “”, line 1006, in _gcd_import
File “”, line 983, in _find_and_load
File “”, line 967, in _find_and_load_unlocked
File “”, line 677, in _load_unlocked
File “”, line 728, in exec_module
File “”, line 219, in _call_with_frames_removed
File “/tmp/student_solution_cells/”, line 23, in
File “/tmp/”, line 106, in compute_information_gain_test
assert np.isclose(result, 0, atol=1e-6), f"Wrong information gain. Expected {0.0} got: {result}"
AssertionError: Wrong information gain. Expected 0.0 got: nan

I tried to reset the notebook and reconect the kernell but It’s still not working…

The nan error means not a number.
There is an error in your code.