From 5b341ee666062ae7b4cdef87cfadbc804d4890f1 Mon Sep 17 00:00:00 2001 From: Dmitri Gekhtman <62982571+DmitriGekhtman@users.noreply.github.com> Date: Mon, 13 Jun 2022 11:00:47 -0700 Subject: [PATCH] [KubeRay][Minor][CI] Deflake autoscaling test Minor adjustment to e2e test logic of KubeRay test. --- python/ray/tests/kuberay/test_autoscaling_e2e.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/tests/kuberay/test_autoscaling_e2e.py b/python/ray/tests/kuberay/test_autoscaling_e2e.py index b2e21250f..1eaa5acbb 100644 --- a/python/ray/tests/kuberay/test_autoscaling_e2e.py +++ b/python/ray/tests/kuberay/test_autoscaling_e2e.py @@ -398,7 +398,7 @@ class KubeRayAutoscalingTest(unittest.TestCase): script_name="scale_up_custom.py", head_service=HEAD_SERVICE, ) - assert job_logs == "Submitted custom scale request!\n" + assert "Submitted custom scale request!" in job_logs, job_logs logger.info("Confirming two workers have scaled up.") wait_for_pods(goal_num_pods=3, namespace=RAY_CLUSTER_NAMESPACE)