[KubeRay][Minor][CI] Deflake autoscaling test

Minor adjustment to e2e test logic of KubeRay test.
This commit is contained in:
Dmitri Gekhtman 2022-06-13 11:00:47 -07:00 committed by GitHub
parent 3278763dd7
commit 5b341ee666
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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)