You can paste the autograder output here, but you did that earlier and it didn’t help, right?
How about pasting or posting a screenshot of the actual error that you are getting when you run your notebook. You mentioned that the “one hot” routine passes, but then it fails after that.
Here’s a thread about that type of problem that may be relevant.
Does this error message look familiar:
ValueError: Shape must be rank 1 but is rank 0 for '{{node Reshape}} = Reshape[T=DT_FLOAT, Tshape=DT_INT32](one_hot, Reshape/shape)' with input shapes: [6], [].