hunk.code_window:         ray.shutdown()
       def test_a3c(self):
           check_support(
               "A3C", {
                   "num_workers": 1,
                   "optimizer": {
                       "grads_per_step": 1
                   }
               },
               self.stats,
               check_bounds=True)
       def test_appo(self):
           check_support("APPO", {"num_gpus": 0, "vtrace": False}, self.stats)
           check_support(
               "APPO", {
  
hunk.labels: keep keep keep replace replace replace replace replace replace replace replace replace keep keep keep keep keep
hunk.after_edit:         config = {
               "num_workers": 1,
               "optimizer": {
                   "grads_per_step": 1
               }
           }
           check_support("A3C", config, self.stats, check_bounds=True)
           config["use_pytorch"] = True
           check_support("A3C", config, self.stats, check_bounds=True)
  
hunk.file_path: rllib/tests/test_supported_spaces.py
hunk.type: replace
file: package org.ray.streaming.runtime.transfer;
  
  import java.nio.ByteBuffer;
  
  /**
   * DataMessage represents data between upstream and downstream operator
   */
  public class DataMessage implements Message {
    private final ByteBuffer body;
    private final long msgId;
    private final long timestamp;
    private final String channelId;
  
    public DataMessage(ByteBuffer body, long timestamp, long msgId, String channelId) {
      this.body = body;
      this.timestamp = timestamp;
      this.msgId = msgId;
      this.channelId = channelId;
    }
  
    @Override
    public ByteBuffer body() {
      return body;
    }
  
    @Override
    public long timestamp() {
      return timestamp;
    }
  
    /**
     * @return message id
     */
    public long msgId() {
      return msgId;
    }
  
    /**
     * @return string id of channel where data is coming from
     */
    public String channelId() {
      return channelId;
    }
  
    @Override
    public String toString() {
      return "DataMessage{" +
          "body=" + body +
          ", msgId=" + msgId +
          ", timestamp=" + timestamp +
          ", channelId='" + channelId + '\'' +
          '}';
    }
  }
  