get_tpu_nodes_for_slice#

ray.util.tpu.get_tpu_nodes_for_slice(slice_name: str, nodes: List[Dict[str, Any]] | None = None) → List[Dict[str, Any]][source]#

Returns all alive Ray nodes belonging to the specified TPU slice.

Parameters:

  • slice_name (str) – The TPU slice name to filter by.

  • nodes (List[Dict[str, Any]] | None) – Optional list of Ray node dictionaries. If not provided, it will be fetched via ray.nodes() from GCS.

Returns:

A list of node dictionaries that are alive and belong to the specified TPU slice.

Return type:

List[Dict[str, Any]]

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