fix unit_test

AlibabaPAI · Jan 21, 2025 · 6c1ef67 · 6c1ef67
1 parent 57d2180
commit 6c1ef67
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit_test/global_scheduler/test_manager.py b/tests/unit_test/global_scheduler/test_manager.py
@@ -478,7 +478,7 @@ async def test_pd_disagg_gloal_launch_deployment_and_auto_scale_up_loop(ray_env,
     prefill_instance_ids = []
     decode_instance_ids = []
     for _, instance_handle in curr_instances.items():
-        instance_type = ray.get(instance_handle.is_ready.remote()).instance_type
+        instance_type = ray.get(instance_handle.get_instance_args.remote()).instance_type
         if instance_type == InstanceType.PREFILL:
             num_prefill_instances += 1
             prefill_instance_ids.append(ray.get(instance_handle.get_instance_info.remote()).instance_id)
@@ -509,7 +509,7 @@ async def test_pd_disagg_gloal_launch_deployment_and_auto_scale_up_loop(ray_env,
     num_decode_instances = 0
     decode_instance_ids = []
     for instance_id, instance_handle in curr_instances.items():
-        instance_type = ray.get(instance_handle.is_ready.remote()).instance_type
+        instance_type = ray.get(instance_handle.get_instance_args.remote()).instance_type
         if instance_type == InstanceType.PREFILL:
             num_prefill_instances += 1
         elif instance_type == InstanceType.DECODE: