code: def h2oai_q10(ds: Dataset) -> Dataset:
      # TODO(chengsu): Run this after dataset supports multiple group-by keys.
      # return ds.groupby(["id1", "id2", "id3", "id4", "id5", "id6"])
      #          .aggregate(Count(), Sum("v3"))
      raise NotImplementedError
  
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 11 for calling a global function with example usage: h2oai_q10(ds) without return types