__init__.py — verl/single_controller/ray/__init__.py

文件路径

verl/single_controller/ray/__init__.py

文件概述

ray 子模块的入口文件，从 base.py 导入并暴露所有 Ray 后端相关的类和函数。

关键代码讲解

from .base import (
    RayClassWithInitArgs,
    RayResourcePool,
    RayWorkerGroup,
    ResourcePoolManager,
    SubRayResourcePool,
    create_colocated_worker_cls,
    create_colocated_worker_cls_fused,
)

__all__ = [
    "RayClassWithInitArgs",
    "RayResourcePool",
    "SubRayResourcePool",
    "RayWorkerGroup",
    "ResourcePoolManager",
    "create_colocated_worker_cls",
    "create_colocated_worker_cls_fused",
]

核心类/函数列表

导出名称	说明
RayClassWithInitArgs	Ray Actor 的延迟实例化包装器
RayResourcePool	基于 Ray PlacementGroup 的资源池
SubRayResourcePool	资源池的子集（用于拆分）
RayWorkerGroup	基于 Ray 的 WorkerGroup 实现
ResourcePoolManager	管理多个资源池的高级接口
create_colocated_worker_cls	创建共存 Worker 类（旧版，已弃用）
create_colocated_worker_cls_fused	创建融合 Worker 类（新版）

与其他模块的关系

• 被框架上层代码导入：如 Trainer 通过 from verl.single_controller.ray import RayWorkerGroup 使用
• 依赖 base.py：所有实现都在 base.py 中

小结

纯导出文件，用于简化导入路径。了解导出列表即可。