class_docstr: Wheel bandit environment for 2D contexts
  (see https://arxiv.org/abs/1802.09127).
class_docstr_tok: ["Wheel","bandit","environment","for","2D","contexts","(","see","https",":","\/\/arxiv.org\/abs\/1802.09127",")","."]
code: def _sample_context(self):
      while True:
          state = np.random.uniform(-1, 1, self.feature_dim)
          if np.linalg.norm(state) <= 1:
              return state
  
imports: import copy
  import gymnasium
  from gymnasium.spaces import Box, Discrete
  import numpy
  import random
label_desc: Use image node_id 2 for calling the WheelBanditEnv obj's underlying member method code with example usage: obj._sample_context() and returns: state