code: def mag_to_param_with_op_id(op_id):
      def mag_to_param(magnitude):
          return np.array([op_id, magnitude], dtype=np.int32)
  
      return mag_to_param
  
imports: import itertools
  import os
  import unittest
  import random
  import numpy
  from scipy.stats import chisquare
  from nose2.tools import params
  from nvidia.dali import fn, types
  from nvidia.dali import pipeline_def
  from nvidia.dali.auto_aug import auto_augment, augmentations
  from nvidia.dali.auto_aug.core import augmentation, Policy
  from test_utils import get_dali_extra_path, check_batch
  from nose_utils import assert_raises, assert_warns
label_desc: Use image node_id 1 for calling a global function with example usage: mag_to_param_with_op_id(op_id) and returns: mag_to_param, np