code: def to_jnp_array(data: np.ndarray):
      return jnp.array(data, dtype=jnp.float32)
  
file_path: evojax/evojax/task/procgen_task.py
imports: import sys
  import multiprocessing
  import numpy
  from typing import Tuple
  import jax
  import jax.numpy
  from flax.struct import dataclass
  from evojax.task.base import VectorizedTask
  from evojax.task.base import TaskState
label_desc: Use image node_id 1 for calling a global function with example usage: to_jnp_array(data) and returns: jnp