Deflake cluster heartbeat test (#5552)

python/ray/resource_spec.py

@@ -197,7 +197,7 @@ class ResourceSpec(
         logger.info(
             "Starting Ray with {} GiB memory available for workers and up to "
             "{} GiB for objects. You can adjust these settings "
-            "with ray.remote(memory=<bytes>, "
+            "with ray.init(memory=<bytes>, "
             "object_store_memory=<bytes>).".format(
                 round(
                     ray_constants.round_to_memory_units(

python/ray/tests/test_multi_node_2.py

@@ -2,6 +2,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
+from flaky import flaky
 import logging
 import pytest
 import time
@@ -148,14 +149,15 @@ def test_heartbeats_single(ray_start_cluster_head):
     ray.get(work_handle)
 
 
+@flaky(max_runs=4)
 def test_heartbeats_cluster(ray_start_cluster_head):
     """Unit test for `Cluster.wait_for_nodes`.
 
     Test proper metrics.
     """
     cluster = ray_start_cluster_head
-    timeout = 5
-    num_workers_nodes = 4
+    timeout = 8
+    num_workers_nodes = 3
     num_nodes_total = int(num_workers_nodes + 1)
     [cluster.add_node() for i in range(num_workers_nodes)]
     cluster.wait_for_nodes()