fix compute_test_error

one_hot being deprecated broke this script. This fixes it.