volcengine · eric-haibin-lin · Jan 8, 2025 · Jan 8, 2025 · Jan 8, 2025
diff --git a/examples/ray/tutorial.ipynb b/examples/ray/tutorial.ipynb
@@ -437,7 +437,7 @@
    },
    "outputs": [],
    "source": [
-    "from verl.single_controller.ray.decorator import register, Dispatch, Execute"
+    "from verl.single_controller.base.decorator import register, Dispatch, Execute"
    ]
   },
   {
@@ -518,7 +518,7 @@
    },
    "outputs": [],
    "source": [
-    "from verl.single_controller.ray.decorator import register, Dispatch, collect_all_to_all, Execute"
+    "from verl.single_controller.base.decorator import register, Dispatch, collect_all_to_all, Execute"
    ]
   },
   {
@@ -723,7 +723,7 @@
    },
    "outputs": [],
    "source": [
-    "from verl.single_controller.ray.decorator import register, Dispatch, Execute\n",
+    "from verl.single_controller.base.decorator import register, Dispatch, Execute\n",
     "from verl.single_controller.ray.megatron import NVMegatronRayWorkerGroup\n",
     "from verl.single_controller.base.megatron.worker import MegatronWorker\n",
     "from verl.single_controller.ray.base import RayResourcePool, RayClassWithInitArgs, RayWorkerGroup\n",

diff --git a/tests/ray/detached_worker/server.py b/tests/ray/detached_worker/server.py
@@ -28,7 +28,7 @@
 from verl.single_controller.ray import RayClassWithInitArgs, RayResourcePool
 from verl.single_controller.ray.megatron import NVMegatronRayWorkerGroup
 from verl.single_controller.base.megatron.worker import MegatronWorker
-from verl.single_controller.ray.decorator import register, Dispatch
+from verl.single_controller.base.decorator import register, Dispatch
 from verl import DataProto
 from verl.models.llama.megatron import ParallelLlamaForCausalLMRmPadPP
 

diff --git a/tests/ray/test_ray_local_envs.py b/tests/ray/test_ray_local_envs.py
@@ -19,7 +19,7 @@
 
 from verl.single_controller.ray.base import RayResourcePool, RayClassWithInitArgs, RayWorkerGroup
 from verl.single_controller.base.worker import Worker
-from verl.single_controller.ray.decorator import register, Dispatch, collect_all_to_all, Execute
+from verl.single_controller.base.decorator import register, Dispatch, collect_all_to_all, Execute
 
 
 @ray.remote

diff --git a/tests/ray/test_worker_group_basics.py b/tests/ray/test_worker_group_basics.py
@@ -20,7 +20,7 @@
 
 from verl.single_controller.ray.base import RayResourcePool, RayClassWithInitArgs, RayWorkerGroup
 from verl.single_controller.base.worker import Worker
-from verl.single_controller.ray.decorator import register, Dispatch, collect_all_to_all, Execute
+from verl.single_controller.base.decorator import register, Dispatch, collect_all_to_all, Execute
 
 
 def two_to_all_dispatch_fn(worker_group, *args, **kwargs):

diff --git a/verl/single_controller/base/dp.py b/verl/single_controller/base/dp.py
diff --git a/verl/single_controller/ray/decorator.py b/verl/single_controller/ray/decorator.py
diff --git a/verl/single_controller/ray/dist_data_pass_protocol.py b/verl/single_controller/ray/dist_data_pass_protocol.py
diff --git a/verl/single_controller/ray/dp.py b/verl/single_controller/ray/dp.py