code: def temp_files():
      # dist_init needs 2 files + 3 files for before state, after state, in_data.
      with temp_files_ctx(5) as files:
          yield files
  
file_path: fairscale/tests/nn/data_parallel/test_fsdp_shared_weights_mevo.py
imports: from copy import deepcopy
  import pytest
  import torch
  from torch import nn
  import torch.multiprocessing
  from torch.optim import SGD
  from fairscale.experimental.nn import MEVO
  from fairscale.fair_dev.testing.testing import dist_init, in_circle_ci, objects_are_equal, skip_if_single_gpu, 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