diff --git a/ci/long_running_tests/workloads/actor_deaths.py b/ci/long_running_tests/workloads/actor_deaths.py index f3fd71949..5c03e34d0 100644 --- a/ci/long_running_tests/workloads/actor_deaths.py +++ b/ci/long_running_tests/workloads/actor_deaths.py @@ -15,7 +15,7 @@ num_nodes = 2 message = ("Make sure there is enough memory on this machine to run this " "workload. We divide the system memory by 2 to provide a buffer.") assert (num_nodes * object_store_memory + num_redis_shards * redis_max_memory < - ray.utils.get_system_memory() / 2) + ray.utils.get_system_memory() / 2), message # Simulate a cluster on one machine. diff --git a/ci/long_running_tests/workloads/many_drivers.py b/ci/long_running_tests/workloads/many_drivers.py index 8154186cb..f8a0d04d3 100644 --- a/ci/long_running_tests/workloads/many_drivers.py +++ b/ci/long_running_tests/workloads/many_drivers.py @@ -14,7 +14,7 @@ num_nodes = 4 message = ("Make sure there is enough memory on this machine to run this " "workload. We divide the system memory by 2 to provide a buffer.") assert (num_nodes * object_store_memory + num_redis_shards * redis_max_memory < - ray.utils.get_system_memory() / 2) + ray.utils.get_system_memory() / 2), message # Simulate a cluster on one machine. diff --git a/ci/long_running_tests/workloads/many_tasks.py b/ci/long_running_tests/workloads/many_tasks.py index 3daba67ea..562d2112e 100644 --- a/ci/long_running_tests/workloads/many_tasks.py +++ b/ci/long_running_tests/workloads/many_tasks.py @@ -15,7 +15,7 @@ num_nodes = 10 message = ("Make sure there is enough memory on this machine to run this " "workload. We divide the system memory by 2 to provide a buffer.") assert (num_nodes * object_store_memory + num_redis_shards * redis_max_memory < - ray.utils.get_system_memory() / 2) + ray.utils.get_system_memory() / 2), message # Simulate a cluster on one machine. diff --git a/ci/long_running_tests/workloads/many_tasks_serialized_ids.py b/ci/long_running_tests/workloads/many_tasks_serialized_ids.py index 02830b065..27a42ce10 100644 --- a/ci/long_running_tests/workloads/many_tasks_serialized_ids.py +++ b/ci/long_running_tests/workloads/many_tasks_serialized_ids.py @@ -18,7 +18,7 @@ num_nodes = 10 message = ("Make sure there is enough memory on this machine to run this " "workload. We divide the system memory by 2 to provide a buffer.") assert (num_nodes * object_store_memory + num_redis_shards * redis_max_memory < - ray.utils.get_system_memory() / 2) + ray.utils.get_system_memory() / 2), message # Simulate a cluster on one machine.