code: def create(space, vocab_size):
      """Creates a SpaceSerializer for the given Gym space."""
      return {
          gym.spaces.Box: BoxSpaceSerializer,
          gym.spaces.Discrete: DiscreteSpaceSerializer,
          gym.spaces.MultiDiscrete: MultiDiscreteSpaceSerializer,
      }[type(space)](space, vocab_size)
  
imports: import copy
  from absl import logging
  import gin
  import gym
  from jax import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: create(space, vocab_size) and returns: dict