hunk.code_window:                         elif o_name in ["vector", "vector2"]:
                               assert isinstance(a.get_policy().model, FCNetV2)
                       a.train()
                       covered_a.add(a_name)
                       covered_o.add(o_name)
               except UnsupportedSpaceException:
  
hunk.labels: keep replace keep keep keep keep
hunk.after_edit:                             if torch:
                                   assert isinstance(
                                       a.get_policy().model, TorchFCNetV2)
                               else:
                                   assert isinstance(
                                       a.get_policy().model, FCNetV2)
  
hunk.file_path: rllib/tests/test_supported_spaces.py
hunk.type: replace
file: package org.ray.streaming.api.function;
  
  import java.io.Serializable;
  
  /**
   * Interface of streaming functions.
   */
  public interface Function extends Serializable {
  
  }
  