The name remove is highly confusing, and IMO should be renamed. Some examples could be:
- finalize / finalize_pruning
- commit / commit_pruning
- consolidate / consolidate_pruning
- persist
- bake
another option could be make_pruning_permanent, or, at the very least, call it something like remove_reparameterization