From dac967a59b398611aa3285ee63bbc029cd0aef5e Mon Sep 17 00:00:00 2001 From: Srihari Date: Tue, 19 Nov 2024 12:13:45 +0530 Subject: [PATCH] Fix upgrade tests Ray Job Memory error for 2.11 release branch --- tests/upgrade/raycluster_sdk_upgrade_test.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/upgrade/raycluster_sdk_upgrade_test.py b/tests/upgrade/raycluster_sdk_upgrade_test.py index 235383f6..8098944d 100644 --- a/tests/upgrade/raycluster_sdk_upgrade_test.py +++ b/tests/upgrade/raycluster_sdk_upgrade_test.py @@ -49,12 +49,11 @@ def run_mnist_raycluster_sdk_oauth(self): namespace=self.namespace, num_workers=1, head_cpus=1, - head_memory=2, + head_memory=4, min_cpus=1, max_cpus=1, - min_memory=1, - max_memory=2, - num_gpus=0, + min_memory=4, + max_memory=4, image=ray_image, write_to_file=True, verify_tls=False,