code: def generate_random_image(image_num: int) -> str:
      wandb_env_vars = {
          k: v for k, v in os.environ.items() if k.startswith("WANDB")
      }
      print(f"wandb_env_vars: {wandb_env_vars}")
  
      run = wandb.init(project="ray-prefect")
      pixels = np.random.randint(low=0, high=256, size=(100, 100, 3))
  
      means = np.zeros((10, 10, 3))
      for i in range(10):
          for j in range(10):
              for c in range(3):
                  means[i, j, c] = np.mean(
                      pixels[
                          i * 10 : (i + 1) * 10,
                          j * 10 : (j + 1) * 10,
                          c,
                      ]
                  )
  
      for i in range(10):
          for j in range(10):
              for c in range(3):
                  pixels[
                      i * 10 : (i + 1) * 10, j * 10 : (j + 1) * 10, c
                  ] = means[i, j, c]
  
      # save as png with name _junk/i.png and pad i to 2 digits
      save_path = DATA_PATH / f"{image_num:02d}.png"
      image = pixels.astype(np.uint8)
      plt.imsave(save_path, image)
  
      run.log({"image": wandb.Image(image)})
  
      # fig = plt.figure()
      # ax = fig.add_subplot(111)
      # ax.imshow(pixels)
      # plt.show()
  
      run.finish()
  
      return str(save_path)
  
file_path: wandb/tests/functional_tests/t0_main/prefect_ray/t1_prefect_ray_workflow.py
imports: import argparse
  import os
  import pathlib
  from typing import List
  import matplotlib.pyplot
  import numpy
  import prefect
  import prefect.task_runners
  import prefect_ray
  import wandb
label_desc: Use image node_id 1 for calling a global function with example usage: generate_random_image(image_num) and returns: str