code: def h2oai_q1(ds: Dataset) -> Dataset:
      return ds.groupby("id1").sum("v1")
  
file_path: ray/release/nightly_tests/dataset/aggregate_benchmark.py
imports: from typing import Tuple
  import ray
  from ray.data.aggregate import Max, Mean, Min, Sum
  from ray.data.aggregate._aggregate import _AggregateOnKeyBase
  from ray.data.block import Block
  from ray.data.dataset import Dataset
  import pyarrow.compute
  from benchmark import Benchmark
label_desc: Use image node_id 2 for calling a global function with example usage: h2oai_q1(ds) and returns: ds