docstr: Returns a config for a test experiment with the given number of steps.
docstr_tok: ["Returns","a","config","for","a","test","experiment","with","the","given","number","of","steps","."]
code: def _get_experiment_config(
      self, *, num_train_steps: int
  ) -> experiments.ExperimentConfig:
      """Returns a config for a test experiment with the given number of steps."""
  
      def environment_factory(seed: int) -> dm_env.Environment:
          del seed
          return fakes.ContinuousEnvironment(
              episode_length=10, action_dim=3, observation_dim=5
          )
  
      num_train_steps = 100
  
      sac_config = sac.SACConfig()
      checkpointing_config = experiments.CheckpointingConfig(
          directory=self.get_tempdir(), time_delta_minutes=0
      )
      return experiments.ExperimentConfig(
          builder=sac.SACBuilder(sac_config),
          environment_factory=environment_factory,
          network_factory=sac.make_networks,
          seed=0,
          max_num_actor_steps=num_train_steps,
          checkpointing=checkpointing_config,
      )
  
file_path: acme/acme/jax/experiments/run_experiment_test.py
imports: from acme.agents.jax import sac
  from acme.jax import experiments
  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 RunExperimentTest obj's underlying member method code with example usage: obj._get_experiment_config() and returns: experiments, fakes