From 032e8553c78d9f72db99be6f95621887a4d24d7f Mon Sep 17 00:00:00 2001 From: Edward Oakes Date: Wed, 11 Dec 2019 17:53:30 -0800 Subject: [PATCH] use numpy in long-running tests (#6448) --- ci/long_running_tests/workloads/many_actor_tasks.py | 3 +++ ci/long_running_tests/workloads/many_tasks.py | 4 +++- 2 files changed, 6 insertions(+), 1 deletion(-) 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