code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--pop-size",
          type=int,
          default=1024,
          help="NE population size.",
      )
      parser.add_argument(
          "--num-tests",
          type=int,
          default=128,
          help="Number of test rollouts.",
      )
      parser.add_argument(
          "--n-repeats",
          type=int,
          default=8,
          help="Training repetitions.",
      )
      parser.add_argument(
          "--max-iter",
          type=int,
          default=300,
          help="Max training iterations.",
      )
      parser.add_argument(
          "--test-interval", type=int, default=50, help="Test interval."
      )
      parser.add_argument(
          "--log-interval",
          type=int,
          default=10,
          help="Logging interval.",
      )
      parser.add_argument(
          "--seed",
          type=int,
          default=42,
          help="Random seed for training.",
      )
      parser.add_argument(
          "--iso-sigma", type=float, default=0.05, help="Iso sigma."
      )
      parser.add_argument(
          "--line-sigma", type=float, default=0.3, help="Line sigma."
      )
      parser.add_argument("--gpu-id", type=str, help="GPU(s) to use.")
      parser.add_argument(
          "--debug", action="store_true", help="Debug mode."
      )
      parser.add_argument(
          "--save-gif", action="store_true", help="Save some GIFs."
      )
      config, _ = parser.parse_known_args()
      return config
  
file_path: evojax/examples/train_ant_map_elites.py
imports: import argparse
  import os
  import numpy
  import matplotlib.pyplot
  from PIL import Image
  from functools import partial
  from evojax import Trainer
  from evojax.task.brax_task import BraxTask
  from evojax.task.brax_task import AntBDExtractor
  from evojax.policy import MLPPolicy
  from evojax.algo import MAPElites
  from evojax import util
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: config