code: def step_stat(p, step=0, **kwargs):
      "Register the number of steps done in `state` for `p`"
      step += 1
      return {"step": step}
  
imports: from __future__ import annotations
  from .torch_basics import *
label_desc: Use image node_id 11 for calling a global function with example usage: step_stat(p, step) and returns: dict