class_docstr: This is just the gym environment, without extra functionality
class_docstr_tok: ["This","is","just","the","gym","environment",",","without","extra","functionality"]
docstr: State comprises of:
  - initial mus
  - legal_indices mask
  - randomly-generated mu changes
docstr_tok: ["State","comprises","of",":","-","initial","mus","-","legal_indices","mask","-","randomly-generated","mu","changes"]
code: def state(self):
      """
      State comprises of:
      - initial mus
      - legal_indices mask
      - randomly-generated mu changes
      """
      self.mu_changes = get_mu_changes(self.num_arms)
      legal_indices_mask = (
          get_legal_indices_mask(self.num_arms)
          & self.legal_indices_mask
      )
      self.legal_indices = legal_indices_mask.nonzero(as_tuple=True)[0]
      result = torch.stack(
          [self.mus, legal_indices_mask, self.mu_changes]
      )
      return result.numpy()
  
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 5 for calling the ChangingArmsEnv obj's underlying member method code with example usage: obj.state() and returns: result