Gymnasium/gym/wrappers/order_enforcing.py

from typing import Optional

import gym


class OrderEnforcing(gym.Wrapper):
    def __init__(self, env):
        super().__init__(env)
        self._has_reset = False

    def step(self, action):
        assert self._has_reset, "Cannot call env.step() before calling reset()"
        observation, reward, done, info = self.env.step(action)
        return observation, reward, done, info

    def reset(self, seed: Optional[int] = None, **kwargs):
        self._has_reset = True
        return self.env.reset(seed=seed, **kwargs)