I am getting the error \'Failed to convert a NumPy array to a Tensor (Unsupported object type float).\' on the input of my label to tf.data.Dataset