code: def build_experiment_config():
      """Builds IMPALA experiment config which can be executed in different ways."""
  
      # Create an environment, grab the spec, and use it to create networks.
      env_name = ENV_NAME.value
  
      def env_factory(seed):
          del seed
          return helpers.make_atari_environment(
              level=env_name,
              sticky_actions=True,
              zero_discount_on_life_loss=False,
              oar_wrapper=True,
          )
  
      # Construct the agent.
      num_learner_steps = NUM_ACTOR_STEPS.value // (
          _SEQUENCE_PERIOD * _BATCH_SIZE
      )
      lr_schedule = optax.linear_schedule(2e-4, 0.0, num_learner_steps)
      config = impala.IMPALAConfig(
          batch_size=_BATCH_SIZE,
          sequence_length=_SEQUENCE_LENGTH,
          sequence_period=_SEQUENCE_PERIOD,
          learning_rate=lr_schedule,
          entropy_cost=5e-3,
          max_abs_reward=1.0,
      )
  
      return experiments.ExperimentConfig(
          builder=impala_builder.IMPALABuilder(config),
          environment_factory=env_factory,
          network_factory=impala.make_atari_networks,
          seed=SEED.value,
          max_num_actor_steps=NUM_ACTOR_STEPS.value,
      )
  
file_path: acme/examples/baselines/rl_discrete/run_impala.py
imports: from absl import flags
  from acme.agents.jax import impala
  from acme.agents.jax.impala import builder
  import helpers
  from absl import app
  from acme.jax import experiments
  from acme.utils import lp_utils
  import launchpad
  import optax
label_desc: Use image node_id 1 for calling a global function with example usage: build_experiment_config() and returns: experiments, helpers