def METHOD_NAME(self, unit_ids):
    current_sorting = self._sorting_stages[self._sorting_stages_i]
    unit2keep = [u for u in current_sorting.get_unit_ids() if u not in unit_ids]
    if self._make_graph:
        i = self._sorting_stages_i
        edges = [(node_t(u, i), node_t(u, i + 1)) for u in unit2keep]
    else:
        edges = None
    self._add_new_stage(current_sorting.select_units(unit2keep), edges)