code: def add_wandb_env_variables(op):
      env = {
          "WANDB_API_KEY": os.getenv("WANDB_API_KEY"),
          "WANDB_BASE_URL": os.getenv("WANDB_BASE_URL"),
          "WANDB_KUBEFLOW_URL": os.getenv("WANDB_KUBEFLOW_URL"),
          "WANDB_PROJECT": "wandb_kfp_integration_test",
      }
  
      for name, value in env.items():
          op = op.add_env_variable(V1EnvVar(name, value))
      return op
  
file_path: wandb/tests/functional_tests/t0_main/kfp/kfp-pipeline-helper.py
imports: import os
  import random
  from typing import NamedTuple
  import kfp
  import kfp.dsl
  from kfp import components
  from kubernetes.client.models import V1EnvVar
  from wandb_probe import wandb_probe_package
label_desc: Use image node_id 1 for calling a global function with example usage: add_wandb_env_variables(op) and returns: op