@jnothman I hate asking this because it might mean additional complexity, but would it make sense to have an argument, e.g. `reset_all=True` that controls which modules are considered for clearing?