code: def __init__(self, env):
      self.env = env
      env_info = env.get_env_info()
      self.episode_limit = env_info["episode_limit"]
      self.n_actions = env_info["n_actions"]
      self.n_agents = env_info["n_agents"]
      self.state_shape = env_info["state_shape"]
      self.obs_shape = (
          env_info["obs_shape"] + self.n_agents + self.n_actions
      )
      self.agent_id_one_hot_transform = OneHotTransform(self.n_agents)
      self.actions_one_hot_transform = OneHotTransform(self.n_actions)
      self._init_agents_id_one_hot(self.n_agents)
  
file_path: PARL/examples/QMIX/env_wrapper.py
imports: from utils import OneHotTransform
  from copy import deepcopy
  import numpy
label_desc: Use image node_id 1 to create a new SC2EnvWrapper object from inherited base classes: object with example: obj = SC2EnvWrapper(env)