[Test] Fix test event test timeout (ray-project#33704)

Signed-off-by: elliottower <[email protected]>
elliottower · Apr 22, 2023 · b39dc58 · b39dc58
1 parent b21ab87
commit b39dc58
Show file tree

Hide file tree

Showing 4 changed files with 658 additions and 623 deletions.
diff --git a/dashboard/modules/actor/actor_head.py b/dashboard/modules/actor/actor_head.py
@@ -194,11 +194,12 @@ def process_actor_data_from_pubsub(actor_id, actor_table_data):
                 logger.debug(
                     f"Processing takes {elapsed}. Total process: " f"{len(published)}"
                 )
-                logger.debug(
-                    "Processing throughput: "
-                    f"{self.total_published_events / self.accumulative_event_processing_s}"  # noqa
-                    " / s"
-                )
+                if self.accumulative_event_processing_s > 0:
+                    logger.debug(
+                        "Processing throughput: "
+                        f"{self.total_published_events / self.accumulative_event_processing_s}"  # noqa
+                        " / s"
+                    )
                 logger.debug(f"queue size: {self.subscriber_queue_size}")
             except Exception:
                 logger.exception("Error processing actor info from GCS.")

diff --git a/python/ray/tests/BUILD b/python/ray/tests/BUILD
@@ -126,6 +126,7 @@ py_test_module_list(
     "test_nested_task.py",
     "test_metrics.py",
     "test_task_events.py",
+    "test_task_events_2.py",
     "test_task_metrics.py",
     "test_task_metrics_reconstruction.py",
     "test_actor_state_metrics.py",