code: def get_rewards(
      agents: BubbleStatus, items: BubbleStatus
  ) -> Tuple[BubbleStatus, BubbleStatus, jnp.ndarray]:
      return get_reward(agents, items)
  
imports: from typing import Tuple
  from functools import partial
  from PIL import Image
  from PIL import ImageDraw
  import numpy
  import jax
  import jax.numpy
  from jax import random
  from jax import tree_util
  from flax.struct import dataclass
  from evojax.task.base import TaskState
  from evojax.task.base import VectorizedTask
label_desc: Use image node_id 3 for calling a global function with example usage: get_rewards(agents, items) and returns: get_reward