def get_action_space(environment_spec):
  """Get action space associated with environment spec.

  Args:
     environment_spec:  EnvironmentSpec object

  Returns:
    OpenAi Gym action space
  """
  return environment_spec.env_lambda().action_space