code: def save_lattices(
      log_dir: str,
      file_name: str,
      fitness_lattice: jnp.ndarray,
      params_lattice: jnp.ndarray,
      occupancy_lattice: jnp.ndarray,
  ) -> None:
      """Save QD method's lattices."""
      file_name = os.path.join(log_dir, "{}.npz".format(file_name))
      np.savez(
          file_name,
          params_lattice=np.array(params_lattice),
          fitness_lattice=np.array(fitness_lattice),
          occupancy_lattice=np.array(occupancy_lattice),
      )
  
file_path: evojax/evojax/util.py
imports: import os
  import logging
  import numpy
  from typing import Union
  from typing import Tuple
  from typing import Callable
  import jax.numpy
  from jax import tree_util
  from flax.core import FrozenDict
label_desc: Use image node_id 5 for calling a global function with example usage: save_lattices(log_dir, file_name, fitness_lattice, params_lattice, occupancy_lattice) without return types