code: def wave_pattern(i):
      return np.array(
          [gen_point(theta, chi, i) for [theta, chi] in theta_chi]
      )
  
imports: import os
  from math import cos, pi, sin
  import numpy
  import wandb
label_desc: Use image node_id 2 for calling a global function with example usage: wave_pattern(i) and returns: np