Change tests.

apache · Aug 13, 2019 · f03214e · f03214e
1 parent 979002e
commit f03214e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/python/unittest/test_numpy_op.py b/tests/python/unittest/test_numpy_op.py
@@ -230,8 +230,9 @@ def hybrid_forward(self, F, a):
     ]
 
     for hybridize in [True, False]:
-        for a_shape in tensor_shapes:
+        for shape in tensor_shapes:
             for dtype in [_np.float32, _np.float64]:
+                a_shape = (1,) + shape
                 test_det = TestDet()
                 if hybridize:
                     test_det.hybridize()