diff --git a/ci/long_running_tests/workloads/many_actor_tasks.py b/ci/long_running_tests/workloads/many_actor_tasks.py index 04c49f351..e4b17dce9 100644 --- a/ci/long_running_tests/workloads/many_actor_tasks.py +++ b/ci/long_running_tests/workloads/many_actor_tasks.py @@ -6,6 +6,8 @@ from __future__ import print_function import time +import numpy as np + import ray from ray.cluster_utils import Cluster @@ -43,6 +45,7 @@ class Actor(object): def method(self): self.value += 1 + return np.zeros(1024, dtype=np.uint8) actors = [ diff --git a/ci/long_running_tests/workloads/many_tasks.py b/ci/long_running_tests/workloads/many_tasks.py index c46826835..0f05b9826 100644 --- a/ci/long_running_tests/workloads/many_tasks.py +++ b/ci/long_running_tests/workloads/many_tasks.py @@ -6,6 +6,8 @@ from __future__ import print_function import time +import numpy as np + import ray from ray.cluster_utils import Cluster @@ -38,7 +40,7 @@ ray.init(address=cluster.address) @ray.remote def f(*xs): - return 1 + return np.zeros(1024, dtype=np.uint8) iteration = 0