Remove unused one_hot argument