Commit 7f477b07 authored by salbiach's avatar salbiach
Browse files

simplified code

parent 0ed10926
......@@ -48,9 +48,8 @@ def block_shuffle(x, y):
@task(
x={Type: COLLECTION_IN, Depth: 2},
y={Type: COLLECTION_IN, Depth: 2},
parameters=IN,
returns=(np.array, np.array))
def block_shuffle_async(x, y, parameters):
def block_shuffle_async(x, y):
x = Array._merge_blocks(x)
y = Array._merge_blocks(y)
p = np.random.permutation(x.shape[0])
......
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