def set_task_options(task: ray_client_pb2.ClientTask, options: Optional[Dict[str, Any]], field: str = "options") -> None: if options is None: task.ClearField(field) return getattr(task, field).pickled_options = pickle.dumps(options)
def set_task_options(task: ray_client_pb2.ClientTask, options: Optional[Dict[str, Any]], field: str = "options") -> None: if options is None: task.ClearField(field) return options_str = json.dumps(options) getattr(task, field).json_options = options_str
def set_task_options(task: ray_client_pb2.ClientTask, options: Optional[Dict[str, Any]], field: str = "options") -> None: if options is None: task.ClearField(field) return # If there's a non-null "placement_group" in `options`, convert the # placement group to a dict so that `options` can be passed to json.dumps. if options.get("placement_group", None): options["placement_group"] = options["placement_group"].to_dict() options_str = json.dumps(options) getattr(task, field).json_options = options_str