def reset(self, rng):
    super().reset(rng)
    self._sls_impl.open()
    self._midway_infeasible = 0
def reset(self, rng):
    super().reset(rng)
    self._sls_impl.open()
    self._midway_infeasible = 0