code: def numerical_data_type():
      dtypes = [torch.FloatTensor, torch.DoubleTensor]
      if TEST_ON_GPU:
          dtypes += [torch.cuda.FloatTensor, torch.cuda.DoubleTensor]
      return dtypes
  
imports: import itertools
  import os
  import numpy
  import pytest
  import torch
  import networkx
  import bluefog.torch
  from common import mpi_env_rank_and_size
label_desc: Use image node_id 2 for calling a global function with example usage: numerical_data_type() and returns: dtypes