docstr: Args:
      input_space: environment space
  
  Returns:
      space shape
docstr_tok: ["Args",":","input_space",":","environment","space","Returns",":","space","shape"]
code: def get_shape(self, input_space):
      """
      Args:
          input_space: environment space
  
      Returns:
          space shape
      """
      if isinstance(input_space, spaces.Box):
          if len(input_space.shape) == 1:
              return input_space.shape[0]
          else:
              return input_space.shape
      elif isinstance(input_space, spaces.Discrete):
          return input_space.n
      else:
          print(
              "[Error] shape is {}, not Box or Discrete".format(
                  input_space.shape
              )
          )
          raise NotImplementedError
  
file_path: PARL/parl/env/multiagent_env.py
imports: import numpy
label_desc: Use image node_id 2 for calling the mpe_wrapper_for_pettingzoo obj's underlying member method code with example usage: obj.get_shape(input_space) and returns: input_space, input_space, input_space