Merge branch 'master' into claude/ray-issue-51442-c7b3c7

Author: chenyuan99

.github/CODEOWNERS

@@ -21,7 +21,7 @@
 /python/ray/ @ray-project/ray-core
 
 # Java worker.
-/java/ @kfstorm @raulchen @WangTaoTheTonic @SongGuyang @ray-project/ray-core
+/java/ @ray-project/ray-core
 
 # C++ worker
 /cpp/ @SongGuyang @raulchen @kfstorm @ray-project/ray-core
@@ -32,7 +32,7 @@
 /doc/source/cluster/kubernetes/ @andrewsykim @ray-project/ray-core @ray-project/ray-docs
 
 # Public protobuf files.
-/src/ray/protobuf/public/ @edoakes @dayshah @MengjinYan
+/src/ray/protobuf/public/ @edoakes @MengjinYan
 
 # Azure autoscaler
 /python/ray/autoscaler/azure/ @ray-project/ray-core @marosset @jackfrancis @alimaazamat
@@ -45,7 +45,7 @@
 
 # Common directory shared by core and the libraries.
 # Stricter enforcement because this defines the public boundary supported by core.
-/python/ray/_common/ @edoakes @dayshah @MengjinYan
+/python/ray/_common/ @edoakes @MengjinYan
 
 # Ray data.
 /python/ray/data/ @ray-project/ray-data

doc/source/template_pins.json

@@ -2,7 +2,7 @@
   "_note": "Pinned templates.ci.ray.io build ids for the docs build. Managed by the auto-bump workflow in anyscale/docs; prefer bumping via that PR rather than hand-editing. Consumed by template_collections.py. 'pins' maps each template in _TEMPLATE_COLLECTIONS to a build id, fetched as /templates/<name>/<id>/build.zip.",
   "pins": {
     "asynchronous_inference": "20260709-002741",
-    "audio-dataset-curation-llm-judge": "20260608-222228",
+    "audio-dataset-curation-llm-judge": "20260709-175135",
     "deepspeed_finetune": "20260709-002714",
     "deployment-serve-llm": "20260709-065934",
     "distributing-pytorch": "20260709-055846",

python/ray/data/_internal/execution/operators/map_transformer.py

@@ -101,18 +101,18 @@ def __init__(
         *,
         is_udf: bool = False,
         output_block_size_option: Optional[OutputBlockSizeOption] = None,
-        reports_custom_op_stats: bool = False,
+        should_report_custom_op_stats: bool = False,
     ):
         """Initialize a :class:`MapTransformFn`.
 
         Args:
             fn: The wrapped transform callable. Invoked with ``(data, ctx)``, or
                 ``(data, ctx, report_custom_op_stats)`` when
-                ``reports_custom_op_stats=True``.
+                ``should_report_custom_op_stats=True``.
             input_type: Expected type of the input data.
             is_udf: Whether this transformation is UDF or not.
             output_block_size_option: (Optional) Output block size configuration.
-            reports_custom_op_stats: If ``True``, the wrapped callable accepts a
+            should_report_custom_op_stats: If ``True``, the wrapped callable accepts a
                 third ``report_custom_op_stats`` callback argument and may report
                 per-task :class:`CustomOpStats` to the driver. Defaults to
                 ``False``, in which case the callable is invoked with
@@ -122,7 +122,7 @@ def __init__(
         self._input_type = input_type
         self._output_block_size_option = output_block_size_option
         self._is_udf = is_udf
-        self._reports_custom_op_stats = reports_custom_op_stats
+        self._should_report_custom_op_stats = should_report_custom_op_stats
 
     @abstractmethod
     def _post_process(self, results: Iterable[MapTransformFnData]) -> Iterable[Block]:
@@ -137,10 +137,10 @@ def _apply_transform(
         """Call the wrapped fn, passing ``report_custom_op_stats`` only if it opted in.
 
         Keeps the common ``(data, ctx)`` signature for the vast majority of
-        transforms; only those constructed with ``reports_custom_op_stats=True``
+        transforms; only those constructed with ``should_report_custom_op_stats=True``
         receive the report callback.
         """
-        if self._reports_custom_op_stats:
+        if self._should_report_custom_op_stats:
             return self._fn(inputs, ctx, report_custom_op_stats)
         return self._fn(inputs, ctx)
 
@@ -375,14 +375,14 @@ def __init__(
         *,
         is_udf: bool = False,
         output_block_size_option: OutputBlockSizeOption,
-        reports_custom_op_stats: bool = False,
+        should_report_custom_op_stats: bool = False,
     ):
         super().__init__(
             row_fn,
             input_type=MapTransformFnDataType.Row,
             is_udf=is_udf,
             output_block_size_option=output_block_size_option,
-            reports_custom_op_stats=reports_custom_op_stats,
+            should_report_custom_op_stats=should_report_custom_op_stats,
         )
 
     def _pre_process(self, blocks: Iterable[Block]) -> Iterable[MapTransformFnData]:
@@ -438,14 +438,14 @@ def __init__(
         zero_copy_batch: bool = True,
         output_block_size_option: Optional[OutputBlockSizeOption] = None,
         target_batch_size_bytes: int = _DEFAULT_BATCH_SIZE_BYTES,
-        reports_custom_op_stats: bool = False,
+        should_report_custom_op_stats: bool = False,
     ):
         super().__init__(
             batch_fn,
             input_type=MapTransformFnDataType.Batch,
             is_udf=is_udf,
             output_block_size_option=output_block_size_option,
-            reports_custom_op_stats=reports_custom_op_stats,
+            should_report_custom_op_stats=should_report_custom_op_stats,
         )
 
         self._batch_size = batch_size
@@ -487,7 +487,7 @@ def __init__(
         is_udf: bool = False,
         disable_block_shaping: bool = False,
         output_block_size_option: Optional[OutputBlockSizeOption] = None,
-        reports_custom_op_stats: bool = False,
+        should_report_custom_op_stats: bool = False,
     ):
         """
         Initializes the object with a transformation function, accompanying options, and
@@ -500,7 +500,7 @@ def __init__(
             disable_block_shaping: Disables block-shaping, making transformer to
                 produce blocks as is.
             output_block_size_option: (Optional) Configure output block sizing.
-            reports_custom_op_stats: If ``True``, ``block_fn`` accepts a third
+            should_report_custom_op_stats: If ``True``, ``block_fn`` accepts a third
                 ``report_custom_op_stats`` callback argument and may report
                 per-task :class:`CustomOpStats` to the driver.
         """
@@ -510,7 +510,7 @@ def __init__(
             input_type=MapTransformFnDataType.Block,
             is_udf=is_udf,
             output_block_size_option=output_block_size_option,
-            reports_custom_op_stats=reports_custom_op_stats,
+            should_report_custom_op_stats=should_report_custom_op_stats,
         )
 
         self._disable_block_shaping = disable_block_shaping

python/ray/data/tests/test_stats.py

@@ -207,7 +207,9 @@ def set_stats(blocks, ctx, report_custom_op_stats):
     transformer = MapTransformer(
         [
             BlockMapTransformFn(
-                set_stats, disable_block_shaping=True, reports_custom_op_stats=True
+                set_stats,
+                disable_block_shaping=True,
+                should_report_custom_op_stats=True,
             )
         ]
     )
@@ -257,7 +259,9 @@ def set_stats(blocks, ctx, report_custom_op_stats):
     transformer = MapTransformer(
         [
             BlockMapTransformFn(
-                set_stats, disable_block_shaping=True, reports_custom_op_stats=True
+                set_stats,
+                disable_block_shaping=True,
+                should_report_custom_op_stats=True,
             )
         ]
     )
@@ -296,7 +300,9 @@ def passthrough(blocks, ctx):
     upstream = MapTransformer(
         [
             BlockMapTransformFn(
-                report_stats, disable_block_shaping=True, reports_custom_op_stats=True
+                report_stats,
+                disable_block_shaping=True,
+                should_report_custom_op_stats=True,
             )
         ]
     )

python/ray/tests/accelerators/test_nvidia_gpu.py

@@ -82,6 +82,8 @@ def test_gpu_info_parsing(patch_mock_pynvml):
         ("NVIDIA H20", "H20"),
         ("NVIDIA B200", "B200"),
         ("NVIDIA B300", "B300"),
+        ("NVIDIA GB200", "GB200"),
+        ("NVIDIA GB300", "GB300"),
         # Consumer GPUs: the regex does not match the mixed-case product line,
         # so we fall back to a hyphen-joined product name.
         ("NVIDIA GeForce RTX 5090", "GeForce-RTX-5090"),

python/ray/util/accelerators/accelerators.py

@@ -12,6 +12,8 @@
 NVIDIA_H20 = "H20"
 NVIDIA_B200 = "B200"
 NVIDIA_B300 = "B300"
+NVIDIA_GB200 = "GB200"
+NVIDIA_GB300 = "GB300"
 NVIDIA_RTX_PRO_6000 = "RTX-PRO-6000"
 INTEL_MAX_1550 = "Intel-GPU-Max-1550"
 INTEL_MAX_1100 = "Intel-GPU-Max-1100"