code: def set_device():
      return torch.device(
          "cuda:0" if torch.cuda.is_available() else "cpu"
      )
  
file_path: NeMo/tests/fixtures/tts.py
imports: import os
  import pytest
  import soundfile
  import torch
  from nemo.collections.asr.parts.utils.manifest_utils import read_manifest
label_desc: Use image node_id 1 for calling a global function with example usage: set_device() and returns: torch