code: def tensor_parallel(request):
      if request.param is not None:
          return request.param
      return int(os.getenv("WORLD_SIZE", "1"))
  
imports: import pytest
  import time
  import os
  import mii
  from types import SimpleNamespace
  from typing import Union
  from deepspeed.launcher.runner import DLTS_HOSTFILE
  import deepspeed.comm
label_desc: Use image node_id 1 for calling a global function with example usage: tensor_parallel(request) and returns: int, request