Where is new_y_test created? It is a separate cell after the test for one_hot_matrix, right? Did that cell throw an error? If so, what was the error?
You might want to have a look at this other recent thread and see if you made the same mistake discussed there.