code: def get_device() -> torch.device:
      """
      Checks if `cuda` is available. If available, use `cuda` by default, else use `cpu`.
  
      Returns
      -------
      Device to be used.
      """
      return torch.device(
          "cuda" if torch.cuda.is_available() else "cpu"
      )
  
imports: import os
  import random
  from typing import List, Dict, Callable, Union, Optional, TYPE_CHECKING
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  from torch.utils.data import DataLoader, Dataset
  from alibi.explainers.backends.cfrl_base import CounterfactualRLDataset
  from alibi.models.pytorch.actor_critic import Actor, Critic
label_desc: Use image node_id 1 for calling a global function with example usage: get_device() and returns: torch