Source code for flatland.ml.wrapped_rail_env

import abc

from flatland.envs.rail_env import RailEnv




class WrappedRailEnv(abc.ABC):
    """
    An API for ml envs wrapping RailEnv.
    """



    @abc.abstractmethod
    def wrap(self) -> RailEnv:
        raise NotImplementedError()