code: def get_ground_truth_assignments_for_zacharys_karate_club() -> (
      jnp.ndarray
  ):
      """Returns ground truth assignments for Zachary's karate club."""
      return jnp.array(
          [
              0,
              0,
              0,
              0,
              0,
              0,
              0,
              0,
              0,
              1,
              0,
              0,
              0,
              0,
              1,
              1,
              0,
              0,
              1,
              0,
              1,
              0,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
              1,
          ]
      )
  
file_path: jraph/jraph/examples/zacharys_karate_club.py
imports: import logging
  from absl import app
  import haiku
  import jax
  import jax.numpy
  import jraph
  import optax
label_desc: Use image node_id 2 for calling a global function with example usage: get_ground_truth_assignments_for_zacharys_karate_club() and returns: jnp