docstr: Identity environment for testing purposes
  
  :param dim: (int) the size of the dimensions you want to learn
  :param ep_length: (int) the length of each episode in timesteps
docstr_tok: ["Identity","environment","for","testing","purposes",":","param","dim",":","(","int",")","the","size","of","the","dimensions","you","want","to","learn",":","param","ep_length",":","(","int",")","the","length","of","each","episode","in","timesteps"]
code: def __init__(self, dim=1, ep_length=100):
      """
      Identity environment for testing purposes
  
      :param dim: (int) the size of the dimensions you want to learn
      :param ep_length: (int) the length of each episode in timesteps
      """
      space = MultiDiscrete([dim, dim])
      super().__init__(ep_length=ep_length, space=space)
  
imports: import numpy
  from typing import Optional
  from gym import Env, Space
  from gym.spaces import Discrete, MultiDiscrete, MultiBinary, Box
label_desc: Use image node_id 1 to create a new IdentityEnvMultiDiscrete object from inherited base classes: IdentityEnv with example: obj = IdentityEnvMultiDiscrete(dim, ep_length)