code: def cuda_total_devices(request):
      old_cuda_total_devices = os.environ.get(
          "CUDA_TOTAL_DEVICES", None
      )
      os.environ["CUDA_TOTAL_DEVICES"] = str(request.param)
      yield
      if old_cuda_total_devices is not None:
          os.environ["CUDA_TOTAL_DEVICES"] = old_cuda_total_devices
      else:
          os.unsetenv("CUDA_TOTAL_DEVICES")
  
imports: import os
  import pytest
  from jina import Flow, Executor, requests, DocumentArray, Document
label_desc: Use image node_id 1 for calling a global function with example usage: cuda_total_devices(request) without return types