fix test

AlibabaPAI · Nov 12, 2024 · f801b01 · f801b01
1 parent a1a7086
commit f801b01
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit_test/global_scheduler/test_dispatch_scheduler.py b/tests/unit_test/global_scheduler/test_dispatch_scheduler.py
@@ -131,7 +131,7 @@ def test_dispatch_rr():
     dispatch_scheduler.instance_num_requests = instance_num_requests
     dispatch_scheduler.instance_info = instance_info_dict
 
-    num_request = 2 * instance_num + 1
+    num_request = 2 * instance_num + 2
     for idx in range(0, num_request):
         instance_id = dispatch_scheduler.dispatch()
         target_instance_id = idx%dispatch_scheduler.num_dispatch_instances
@@ -140,8 +140,8 @@ def test_dispatch_rr():
     for idx in range(instance_num):
         if idx < dispatch_scheduler.num_dispatch_instances:
             dispatch_scheduler.instance_num_requests[f'instance_{idx}'] = \
-                num_request // dispatch_scheduler.num_dispatch_instances + 1 \
+                num_request // dispatch_scheduler.num_dispatch_instances + (1 \
                     if num_request % dispatch_scheduler.num_dispatch_instances > \
-                        idx % dispatch_scheduler.num_dispatch_instances else 0
+                        idx % dispatch_scheduler.num_dispatch_instances else 0)
         else:
             dispatch_scheduler.instance_num_requests[f'instance_{idx}'] = 0