unique pruning on timestep. in case we need it later