ray.util.tpu.get_tpu_slice_name_from_node#

ray.util.tpu.get_tpu_slice_name_from_node(node: Dict[str, Any]) str | None[source]#

Returns the TPU slice name for a given Ray node dictionary.

Parameters:

node – A dictionary representing a Ray node (returned by ray.nodes()).

Returns:

The TPU slice name if the node belongs to a slice, otherwise None.

PublicAPI (alpha): This API is in alpha and may change before becoming stable.