Commit 84f117d1 authored by Theo Steininger's avatar Theo Steininger

Modified generate_data in test/test_distributed_data_object.py in order to comply with numpy 1.12.

parent d1044571
Pipeline #9600 passed with stage
in 9 minutes and 43 seconds
...@@ -145,6 +145,7 @@ def generate_data(global_shape, dtype, distribution_strategy, ...@@ -145,6 +145,7 @@ def generate_data(global_shape, dtype, distribution_strategy,
local_shape[0] * np.ceil(size / 2.)) local_shape[0] * np.ceil(size / 2.))
if number_of_extras > rank//2: if number_of_extras > rank//2:
local_shape[0] += 1 local_shape[0] += 1
local_shape[0] = int(local_shape[0])
local_shape = tuple(local_shape) local_shape = tuple(local_shape)
a = np.arange(np.prod(local_shape)) a = np.arange(np.prod(local_shape))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment