File size: 864 Bytes
8e63d2a
 
48f24b9
 
 
 
8e63d2a
 
 
 
 
 
 
 
 
 
 
 
48f24b9
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
from typing import TypeVar

from gym_microrts.envs.vec_env import (
    MicroRTSGridModeSharedMemVecEnv,
    MicroRTSGridModeVecEnv,
)

MicroRTSGridModeVecEnvCompatSelf = TypeVar(
    "MicroRTSGridModeVecEnvCompatSelf", bound="MicroRTSGridModeVecEnvCompat"
)


class MicroRTSGridModeVecEnvCompat(MicroRTSGridModeVecEnv):
    @property
    def unwrapped(
        self: MicroRTSGridModeVecEnvCompatSelf,
    ) -> MicroRTSGridModeVecEnvCompatSelf:
        return self


MicroRTSGridModeSharedMemVecEnvCompatSelf = TypeVar(
    "MicroRTSGridModeSharedMemVecEnvCompatSelf",
    bound="MicroRTSGridModeSharedMemVecEnvCompat",
)


class MicroRTSGridModeSharedMemVecEnvCompat(MicroRTSGridModeSharedMemVecEnv):
    @property
    def unwrapped(
        self: MicroRTSGridModeSharedMemVecEnvCompatSelf,
    ) -> MicroRTSGridModeSharedMemVecEnvCompatSelf:
        return self