code: def input_data(request):
      shape = ((2, 3), (3, 4))
      return get_data(shape, dtype=request.param)
  
file_path: fairscale/tests/experimental/nn/test_mevo.py
imports: import os
  import pytest
  import torch
  from fairscale.experimental.nn import MEVO
  from fairscale.experimental.nn.mevo import BaselineSoftmaxNllLoss, get_data
  from fairscale.fair_dev.testing.testing import skip_if_no_cuda
label_desc: Use image node_id 1 for calling a global function with example usage: input_data(request) and returns: get_data