one_hot deprecation fix