code: def as_square_shape(edge_len):
      return [edge_len, edge_len]
  
file_path: DALI/dali/test/python/auto_aug/test_select.py
imports: import os
  import numpy
  from nose2.tools import params
  from nvidia.dali import fn, pipeline_def, types
  from nvidia.dali.auto_aug.core import augmentation, select
  from test_utils import get_dali_extra_path, check_batch
label_desc: Use image node_id 4 for calling a global function with example usage: as_square_shape(edge_len) and returns: list