diff --git a/tests/upgrade/raycluster_sdk_upgrade_test.py b/tests/upgrade/raycluster_sdk_upgrade_test.py index ff98e28f6..e5fd18765 100644 --- a/tests/upgrade/raycluster_sdk_upgrade_test.py +++ b/tests/upgrade/raycluster_sdk_upgrade_test.py @@ -52,8 +52,8 @@ def run_mnist_raycluster_sdk_oauth(self): head_memory=2, worker_cpu_requests=1, worker_cpu_limits=1, - worker_memory_requests=1, - worker_memory_limits=2, + worker_memory_requests=4, + worker_memory_limits=4, image=ray_image, write_to_file=True, verify_tls=False,