From 65f7081116569ba198b122b8a07f82df2eaf9532 Mon Sep 17 00:00:00 2001 From: Connor Goggins Date: Mon, 24 Feb 2020 18:17:14 -0800 Subject: [PATCH] Using const instead of literals --- tests/nightly/test_large_array.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/nightly/test_large_array.py b/tests/nightly/test_large_array.py index 6c8a4fdf00fd..2a4149058eb2 100644 --- a/tests/nightly/test_large_array.py +++ b/tests/nightly/test_large_array.py @@ -480,7 +480,7 @@ def check_embedding(): assert out.shape[0] == LARGE_TENSOR_SHAPE assert out.shape[1] == 1 def check_rnn(): - data = nd.random_normal(shape=(2**28, 4, 4)) + data = nd.random_normal(shape=(LARGE_TENSOR_SHAPE/16, 4, 4)) parameters = nd.random_normal(shape=(7,)) state = nd.random_normal(shape=(1, 4, 1)) mode = 'rnn_relu' @@ -490,7 +490,7 @@ def check_rnn(): out = nd.RNN(data=data, parameters=parameters, state=state, mode=mode, state_size=state_size, num_layers=num_layers) - assert out.shape[0] == 268435456 + assert out.shape[0] == LARGE_TENSOR_SHAPE/16 assert out.shape[1] == 4 assert out.shape[2] == 1