code: def setup_function(_):
      # Reset `forward_calls` before each test.
      global forward_calls
      forward_calls = 0
  
file_path: allennlp/tests/nn/checkpoint/fairscale_checkpoint_wrapper_test.py
imports: import torch
  import torch.nn
  from allennlp.common.params import Params
  from allennlp.common.testing import multi_device
  from allennlp.nn.checkpoint.checkpoint_wrapper import CheckpointWrapper
  from allennlp.nn.checkpoint.fairscale_checkpoint_wrapper import FairScaleCheckpointWrapper
label_desc: Use image node_id 1 for calling a global function with example usage: setup_function(_) without return types