code: def get_ones():
      """Return test data with ones tensor"""
      return (
          0,
          [1, 5, 10, 100],
          torch.ones(100),
          [
              torch.tensor(0.01),
              torch.tensor(0.05),
              torch.tensor(0.1),
              torch.tensor(1.0),
          ],
      )
  
file_path: fairscale/tests/experimental/wgit/test_signal_sparsity_profiling.py
imports: import time
  import pytest
  import torch
  from fairscale.experimental.wgit.signal_sparsity_profiling import EnergyConcentrationProfile
  from fairscale.fair_dev.testing.testing import objects_are_equal, skip_if_no_cuda
label_desc: Use image node_id 2 for calling a global function with example usage: get_ones() and returns: int,  list,  torch,  list