code: def world_size(self):
      return min(torch.cuda.device_count(), 2)
  
file_path: apex/tests/L0/run_deprecated/test_deprecated_warning.py
imports: import unittest
  import torch
  import apex
  from apex.transformer.testing.distributed_test_base import NcclDistributedTestBase
label_desc: Use image node_id 1 for calling the TestParallel obj's underlying member method code with example usage: obj.world_size() and returns: min