code: def __iter__(self):
      for _ in range(self.num_obs):
          obs = self.env.get_batch()  # obs is a batch
          action, log_prob = self.agent.act(obs=obs)
          obs = self.env.add_chosen_action_reward(
              chosen_action_idx=action, batch=obs
          )
          yield obs
  
file_path: ReAgent/reagent/evaluation/cb/run_synthetic_bandit.py
imports: import uuid
  from typing import List, Tuple
  import pytorch_lightning
  import torch
  from pytorch_lightning import seed_everything
  from reagent.evaluation.cb.synthetic_contextual_bandit_data import DynamicBanditAgent, DynamicBanditEnv
label_desc: Use image node_id 2 for calling the BanditDataset obj's underlying member method code with example usage: obj.__iter__() without return types