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 reset(self):
      # initialize the distributions
      self.num_steps = 0
      self.mus = get_initial_mus(self.num_arms)
      # these are turned off when an arm has been "exhausted"
      self.legal_indices_mask = torch.tensor([1] * self.num_arms).to(
          torch.uint8
      )
      return self.state
  
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 4 for calling the ChangingArmsEnv obj's underlying member method code with example usage: obj.reset() and returns: self