hunk.code_window:                 else:
                       a = get_agent_class(alg)(config=config, env="stub_env")
                       if alg not in ["DDPG", "ES", "ARS", "SAC"]:
                           if o_name in ["atari", "image"]:
                               assert isinstance(a.get_policy().model,
                                                 VisionNetV2)
                           elif o_name in ["vector", "vector2"]:
  
hunk.labels: keep keep keep keep replace replace keep
hunk.after_edit:                             if torch:
                                   assert isinstance(
                                       a.get_policy().model, TorchVisionNetV2)
                               else:
                                   assert isinstance(
                                       a.get_policy().model, VisionNetV2)
  
hunk.file_path: rllib/tests/test_supported_spaces.py
hunk.type: replace
file: package org.ray.streaming.runtime.config.types;
  
  public enum SlotAssignStrategyType {
  
    /**
     * Resource scheduling strategy based on FF(First Fit) algorithm and pipeline.
     */
    PIPELINE_FIRST_STRATEGY("pipeline_first_strategy", 0);
  
    private String value;
    private int index;
  
    SlotAssignStrategyType(String value, int index) {
      this.value = value;
      this.index = index;
    }
  
    public String getValue() {
      return value;
    }
  }
  