code: def out_of_screen(state: jnp.ndarray) -> jnp.float32:
      x = state[0]
      beyond_boundary_l = jnp.where(x < -CART_X_LIMIT, 1, 0)
      beyond_boundary_r = jnp.where(x > CART_X_LIMIT, 1, 0)
      return jnp.bitwise_or(beyond_boundary_l, beyond_boundary_r)
  
imports: from typing import Tuple
  from PIL import Image
  from PIL import ImageDraw
  import numpy
  import jax
  import jax.numpy
  from jax import random
  from flax.struct import dataclass
  from evojax.task.base import TaskState
  from evojax.task.base import VectorizedTask
label_desc: Use image node_id 6 for calling a global function with example usage: out_of_screen(state) and returns: jnp