code: def _get_step_num(self, action_name: str) -> int:
      if action_name not in self._step_recoding_map:
          self._step_recoding_map[action_name] = 1
      else:
          self._step_recoding_map[action_name] += 1
      return self._step_recoding_map[action_name]
  
file_path: pytorch-lightning/src/lightning/pytorch/profilers/xla.py
imports: import logging
  from typing import Dict
  from typing_extensions import override
  from lightning.fabric.accelerators.xla import _XLA_AVAILABLE
  from lightning.pytorch.profilers.profiler import Profiler
label_desc: Use image node_id 4 for calling the XLAProfiler obj's underlying member method code with example usage: obj._get_step_num(action_name) and returns: self