code: def parse_args():
      parser = argparse.ArgumentParser()
      parser.add_argument(
          "--pop-size",
          type=int,
          default=256,
          help="NE population size.",
      )
      parser.add_argument(
          "--max-iter",
          type=int,
          default=2000,
          help="Max training iterations.",
      )
      parser.add_argument(
          "--test-interval",
          type=int,
          default=100,
          help="Test interval.",
      )
      parser.add_argument(
          "--log-interval",
          type=int,
          default=100,
          help="Logging interval.",
      )
      parser.add_argument(
          "--seed",
          type=int,
          default=42,
          help="Random seed for training.",
      )
      parser.add_argument("--gpu-id", type=str, help="GPU(s) to use.")
      parser.add_argument(
          "--item-csv", type=str, default=None, help="Items csv file."
      )
      parser.add_argument(
          "--cap-csv",
          type=str,
          default=None,
          help="Capacities csv file.",
      )
      parser.add_argument(
          "--use-synthesized-data",
          action="store_true",
          help="Use synthesized data instead of CSV files.",
      )
      parser.add_argument(
          "--debug", action="store_true", help="Debug mode."
      )
      config, _ = parser.parse_known_args()
      return config
  
file_path: evojax/examples/train_mdkp.py
imports: import argparse
  import os
  import sys
  import numpy
  from typing import Tuple
  import jax
  import jax.numpy
  from evojax import Trainer
  from evojax.task.mdkp import MDKP
  from evojax.task.mdkp import TaskState
  from evojax.policy import PolicyNetwork
  from evojax.policy.base import PolicyState
  from evojax.algo import SimpleGA
  from evojax import util
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: config