code: def test_current_device(self):
      config = {
          "training": {"seed": 1, "cudnn_benchmark": False},
          "distributed": {"init_method": None},
      }
      deviceMock = DeviceMock(OmegaConf.create(config))
      deviceMock.configure_seed()
      deviceMock.configure_device()
      device = get_current_device()
      if torch.cuda.is_available():
          self.assertEqual(device, "cuda:0")
      else:
          self.assertEqual(device, torch.device(type="cpu"))
  
file_path: mmf/tests/trainers/test_device.py
imports: import unittest
  import torch
  from mmf.trainers.core.device import TrainerDeviceMixin
  from mmf.utils.general import get_current_device
  from omegaconf import OmegaConf
label_desc: Use image node_id 1 for calling the TestDevice obj's underlying member method code with example usage: obj.test_current_device() without return types