Remove deprecated one_hot argument