code: def _make_environment_factory(
      env_name: str,
  ) -> jax_types.EnvironmentFactory:
      """Returns the environment factory for the given environment."""
  
      def environment_factory(seed: int) -> dm_env.Environment:
          del seed
          return helpers.make_environment(task=env_name)
  
      return environment_factory
  
imports: from typing import Callable, Iterator, Tuple
  from absl import flags
  from acme import specs
  from acme import types
  from acme.agents.jax import actor_core
  from acme.agents.jax import bc
  from acme.datasets import tfds
  import helpers
  from absl import app
  from acme.jax import experiments
  from acme.jax import types
  from acme.jax import utils
  from acme.utils import lp_utils
  import dm_env
  import haiku
  import launchpad
  import numpy
label_desc: Use image node_id 2 for calling a global function with example usage: _make_environment_factory(env_name) and returns: environment_factory, helpers