code: def reset(self):
          
          inds = list(range(self.state.obj_get_positions().shape[0]))
          self._rad_nms = self.state.param_particle_rad(inds)
          self._pos_nms = self.state.param_particle_pos(inds)
          self._initial_rad = np.copy(self.state.state[self._rad_nms])
          self._initial_pos = np.copy(self.state.state[self._pos_nms]).reshape((-1,3))
          self.param_vals[self.rscale_mask] = 0
docs: Resets the initial radii used for updating the particles. Call
          if any of the particle radii or positions have been changed
          external to the augmented state.