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
  
file_path: acme/examples/baselines/imitation/run_iqlearn.py
imports: from typing import Callable, Iterator
  from absl import flags
  from acme import specs
  from acme import types
  from acme.agents.jax import actor_core
  from acme.agents.jax import iq_learn
  from acme.datasets import tfds
  import helpers
  from absl import app
  from acme.jax import experiments
  from acme.jax import types
  from acme.utils import lp_utils
  import dm_env
  import jax
  import launchpad
label_desc: Use image node_id 1 for calling a global function with example usage: _make_environment_factory(env_name) and returns: environment_factory, helpers