code: def main(_):
      config = build_experiment_config()
      if FLAGS.run_distributed:
          program = experiments.make_distributed_experiment(
              experiment=config, num_actors=4
          )
          lp.launch(
              program,
              xm_resources=lp_utils.make_xm_docker_resources(program),
          )
      else:
          experiments.run_experiment(
              experiment=config,
              eval_every=FLAGS.eval_every,
              num_eval_episodes=FLAGS.evaluation_episodes,
          )
  
file_path: acme/examples/baselines/imitation/run_pwil.py
imports: from typing import Sequence
  from absl import flags
  from acme import specs
  from acme.agents.jax import d4pg
  from acme.agents.jax import pwil
  from acme.datasets import tfds
  import helpers
  from absl import app
  from acme.jax import experiments
  from acme.jax import networks
  from acme.jax import utils
  from acme.utils import lp_utils
  import dm_env
  import haiku
  import jax.numpy
  import launchpad
  import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: main(_) without return types