code: def expand_arg_input(input_data: ArgData, arg_data: ArgData):
      """
      Expands the `arg_data` to match the sequence shape of input_data.
      """
      assert arg_data.desc.expandable_prefix in ["F", ""]
      assert len(input_data.data) == len(arg_data.data)
      arg_has_frames = arg_data.desc.expandable_prefix == "F"
      return [
          expand_arg(
              input_data.desc.expandable_prefix,
              arg_has_frames,
              input_batch,
              arg_batch,
          )
          for input_batch, arg_batch in zip(
              input_data.data, arg_data.data
          )
      ]
  
imports: import os
  import random
  import numpy
  from typing import List, Union, Callable, Optional
  from dataclasses import dataclass, field
  from nvidia.dali import pipeline_def
  import nvidia.dali.fn
  from nvidia.dali import types
  import nvidia.dali.tensors
  from test_utils import get_dali_extra_path, check_batch
label_desc: Use image node_id 8 for calling a global function with example usage: expand_arg_input(input_data, arg_data) and returns: unknown