class_docstr: This is just the gym environment, without extra functionality
class_docstr_tok: ["This","is","just","the","gym","environment",",","without","extra","functionality"]
code: def action_space(self):
      # Selecting 0,1,2...,num_arms-1 is selecting an arm.
      # If action is invalid, agent incurs a penalty.
      # If action is valid, action is an idx i, and reward
      # is a sample from ith distribution. At the same time
      # the ith distribution is updated with the changes.
      # Alternatively, can choose NULL (i.e. do-nothing) action
      # if action = num_arms
      return gym.spaces.Discrete(self.num_arms + 1)
  
imports: import random
  import gym
  import numpy
  import reagent.core.types
  import torch
  from reagent.core.dataclasses import dataclass
  from reagent.core.parameters import NormalizationData, NormalizationKey
  from reagent.gym.envs.env_wrapper import EnvWrapper
  from reagent.gym.normalizers import only_continuous_normalizer
label_desc: Use image node_id 7 for calling the ChangingArmsEnv obj's underlying member method code with example usage: obj.action_space() and returns: gym