Skip to content

add ray-related interface

Compare
Choose a tag to compare
@benerain benerain released this 30 Aug 02:39
· 1 commit to master since this release

add some params that could transfer ray-related args:

  • [1] ray cluster address, str, e.g. "ray://xx.xx.xx.xx:xxxx"
  • [2] ray_remote_args,dic, e.g. "{"resources": {"custome_resource": 1} "
    Valid keywords are ['accelerator_type', 'memory', 'name', 'num_cpus', 'num_gpus', 'object_store_memory', 'placement_group', 'placement_group_bundle_index', 'placement_group_capture_child_tasks', 'resources', 'runtime_env', 'scheduling_strategy', '_metadata', 'max_calls', 'max_retries', 'num_returns', 'retry_exceptions']