Hi, @kd_nitish.
Change the function header to:
def one_hot_matrix(label, depth=6):
Do you have the latest version of the notebook? 
Good luck with the assignment 
Hi, @kd_nitish.
Change the function header to:
def one_hot_matrix(label, depth=6):
Do you have the latest version of the notebook? 
Good luck with the assignment 