docstr: Decide which models (arms) to pull.
  
  We want to pull every model at least burn_in times. This provides a significant accuracy
  boost, albeit at the cost of a longer training time in the beginning.
docstr_tok: ["Decide","which","models","(","arms",")","to","pull",".","We","want","to","pull","every","model","at","least","burn_in","times",".","This","provides","a","significant","accuracy","boost",",","albeit","at","the","cost","of","a","longer","training","time","in","the","beginning","."]
code: def _pick_arms(self):
      """Decide which models (arms) to pull.
  
      We want to pull every model at least burn_in times. This provides a significant accuracy
      boost, albeit at the cost of a longer training time in the beginning.
  
      """
      arm_ids = [
          arm_id
          for arm_id in range(len(self.models))
          if self.policy._counts[arm_id] < self.policy.burn_in
      ]
      if arm_ids:
          return arm_ids
      return [self.policy.pull(range(len(self.models)))]
  
file_path: river/river/model_selection/bandit.py
imports: from __future__ import annotations
  from river import bandit, metrics, model_selection
label_desc: Use image node_id 1 for calling the BanditModelSection obj's underlying member method code with example usage: obj._pick_arms() and returns: list, arm_ids