code: def _get_torch_profiler() -> torch.profiler.profile:
      profiler_schedule = torch.profiler.schedule(
          wait=0,
          warmup=1,
          active=1,
      )
      return torch.profiler.profile(
          activities=[ProfilerActivity.CPU],
          schedule=profiler_schedule,
      )
  
file_path: tnt/tests/utils/data/test_profile_dataloader.py
imports: import unittest
  from typing import Iterator
  import torch
  from torch.profiler import ProfilerActivity
  from torchtnt.utils.data.profile_dataloader import profile_dataloader
  from torchtnt.utils.env import init_from_env
label_desc: Use image node_id 1 for calling a global function with example usage: _get_torch_profiler() and returns: torch