code: def _get_experiment_config(
      self, *, num_learner_steps: int
  ) -> experiments.OfflineExperimentConfig:
      def environment_factory(seed: int) -> dm_env.Environment:
          del seed
          return fakes.ContinuousEnvironment(
              episode_length=10, action_dim=3, observation_dim=5
          )
  
      environment = environment_factory(seed=1)
      environment_spec = specs.make_environment_spec(environment)
  
      def demonstration_dataset_factory(
          random_key: jax_types.PRNGKey,
      ) -> Iterator[types.Transition]:
          del random_key
          batch_size = 64
          return fakes.transition_iterator_from_spec(environment_spec)(
              batch_size
          )
  
      crr_config = crr.CRRConfig()
      crr_builder = crr.CRRBuilder(
          crr_config,
          policy_loss_coeff_fn=crr.policy_loss_coeff_advantage_exp,
      )
      checkpointing_config = experiments.CheckpointingConfig(
          directory=self.get_tempdir(), time_delta_minutes=0
      )
      return experiments.OfflineExperimentConfig(
          builder=crr_builder,
          network_factory=crr.make_networks,
          demonstration_dataset_factory=demonstration_dataset_factory,
          environment_factory=environment_factory,
          max_num_learner_steps=num_learner_steps,
          seed=0,
          environment_spec=environment_spec,
          checkpointing=checkpointing_config,
      )
  
file_path: acme/acme/jax/experiments/run_offline_experiment_test.py
imports: from typing import Iterator
  from acme import specs
  from acme import types
  from acme.agents.jax import crr
  from acme.jax import experiments
  from acme.jax import types
  from acme.jax.experiments import test_utils
  from acme.testing import fakes
  from acme.testing import test_utils
  import dm_env
  from absl.testing import absltest
  from absl.testing import parameterized
label_desc: Use image node_id 3 for calling the RunOfflineExperimentTest obj's underlying member method code with example usage: obj._get_experiment_config() and returns: experiments, fakes, fakes