class_docstr: ### Information set tracker
  
  This is a small helper class to track data from information sets
class_docstr_tok: ["#","#","#","Information","set","tracker","This","is","a","small","helper","class","to","track","data","from","information","sets"]
docstr: Track the data from all information sets
docstr_tok: ["Track","the","data","from","all","information","sets"]
code: def __call__(self, info_sets: Dict[str, InfoSet]):
      """
      Track the data from all information sets
      """
      for I in info_sets.values():
          avg_strategy = I.get_average_strategy()
          for a in I.actions():
              tracker.add(
                  {
                      f"strategy.{I.key}.{a}": I.strategy[a],
                      f"average_strategy.{I.key}.{a}": avg_strategy[a],
                      f"regret.{I.key}.{a}": I.regret[a],
                  }
              )
  
imports: from typing import NewType, Dict, List, Callable, cast
  from labml import monit, tracker, logger, experiment
  from labml.configs import BaseConfigs, option
label_desc: Use image node_id 2 for calling the InfoSetTracker obj's underlying member method code with example usage: obj.__call__(info_sets) without return types