File size: 349 Bytes
912c7e2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from abc import ABC, abstractmethod


class BaseEnv(ABC):
    """
    The base abstract class for all envs.
    """

    @abstractmethod
    def reset(self):
        pass

    @abstractmethod
    def reset_rng(self):
        pass

    @abstractmethod
    def step(self, action):
        pass

    @abstractmethod
    def get_obs(self):
        pass