code: def temp_files():
      num = 2  # dist_init needs 2 files
      files = [tempfile.mkstemp()[1] for _ in range(num)]
  
      yield tuple(files)
  
      # temp files could have been removed, so we use rmf.
      for name in files:
          rmf(name)
  
file_path: fairscale/tests/nn/data_parallel/test_fsdp_input.py
imports: import tempfile
  import pytest
  import torch
  from torch.nn import Linear, Module
  from torch.optim import SGD
  from fairscale.fair_dev.testing.testing import dist_init, rmf, skip_if_no_cuda, teardown
  from fairscale.internal import torch_version
  from fairscale.nn.data_parallel import FullyShardedDataParallel
  from fairscale.nn.data_parallel import TrainingState
label_desc: Use image node_id 1 for calling a global function with example usage: temp_files() without return types