File size: 1,022 Bytes
079c32c |
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 34 |
import copy
import numpy as np
from ding.envs.common import EnvElementRunner, EnvElement
from ding.envs.env.base_env import BaseEnv
from .gfootball_obs import PlayerObs, MatchObs
from ding.utils import deep_merge_dicts
class GfootballObsRunner(EnvElementRunner):
def _init(self, cfg, *args, **kwargs) -> None:
# set self._core and other state variable
self._obs_match = MatchObs(cfg)
self._obs_player = PlayerObs(cfg)
self._core = self._obs_player # placeholder
def get(self, engine: BaseEnv) -> dict:
ret = copy.deepcopy(engine._football_obs)
# print(ret, type(ret))
assert isinstance(ret, dict)
match_obs = self._obs_match._to_agent_processor(ret)
players_obs = self._obs_player._to_agent_processor(ret)
return deep_merge_dicts(match_obs, players_obs)
def reset(self) -> None:
pass
# override
@property
def info(self):
return {'match': self._obs_match.info, 'player': self._obs_player.info}
|