hunk.code_window:     ray.shutdown()
       # Test unavailable type
       runtime_env = {"conda": {"dependencies": ["toolz"]}, "eager_install": 123}
       with pytest.raises(TypeError):
           ray.init(runtime_env=runtime_env)
       ray.shutdown()
  
hunk.labels: keep keep keep keep replace keep keep keep
hunk.after_edit:         ray.init(runtime_env=runtime_env_class(**runtime_env))
  
hunk.file_path: python/ray/tests/test_runtime_env_complicated.py
hunk.type: replace
file: from ray.rllib.evaluation.episode import Episode, MultiAgentEpisode
  from ray.rllib.evaluation.rollout_worker import RolloutWorker
  from ray.rllib.evaluation.sample_batch_builder import (
      SampleBatchBuilder,
      MultiAgentSampleBatchBuilder,
  )
  from ray.rllib.evaluation.sampler import SyncSampler, AsyncSampler
  from ray.rllib.evaluation.postprocessing import compute_advantages
  from ray.rllib.evaluation.metrics import collect_metrics
  from ray.rllib.policy.sample_batch import SampleBatch, MultiAgentBatch
  
  __all__ = [
      "RolloutWorker",
      "SampleBatch",
      "MultiAgentBatch",
      "SampleBatchBuilder",
      "MultiAgentSampleBatchBuilder",
      "SyncSampler",
      "AsyncSampler",
      "compute_advantages",
      "collect_metrics",
      "Episode",
      "MultiAgentEpisode",  # Deprecated -> Use `Episode` instead.
  ]
  