code: def temp_files():
      # dist_init needs 2 files
      with temp_files_ctx(2) as files:
          yield files
  
imports: import pytest
  import torch
  from torch.nn import Linear, Module
  from fairscale.fair_dev.testing.testing import dist_init, skip_if_no_cuda, teardown, temp_files_ctx
  from fairscale.nn.data_parallel import FullyShardedDataParallel
label_desc: Use image node_id 1 for calling a global function with example usage: temp_files() without return types