code: def get_data(shape: tuple) -> Tuple[List, List]:
      sample = torch.randn(shape)
      meta_args = [
          ("sample", sample),
      ]
      concrete_args = [("timestep", 50)]
      return meta_args, concrete_args
  
file_path: ColossalAI/tests/test_autochunk/test_autochunk_diffuser/test_autochunk_unet.py
imports: from typing import List, Tuple
  import pytest
  import torch
  from test_autochunk_diffuser_utils import run_test
  from colossalai.autochunk.autochunk_codegen import AUTOCHUNK_AVAILABLE
  from colossalai.testing import clear_cache_before_run, parameterize, spawn
label_desc: Use image node_id 1 for calling a global function with example usage: get_data(shape) and returns: meta_args,  concrete_args