diff --git a/tests/python/unittest/test_numpy_interoperability.py b/tests/python/unittest/test_numpy_interoperability.py index 05e7343ba0cd..088306ff01c8 100644 --- a/tests/python/unittest/test_numpy_interoperability.py +++ b/tests/python/unittest/test_numpy_interoperability.py @@ -1230,7 +1230,6 @@ def get_mat(n): OpArgMngr.add_workload('diagflat', vals_f, k=-2) - def _add_workload_shape(): OpArgMngr.add_workload('shape', np.random.uniform(size=())) OpArgMngr.add_workload('shape', np.random.uniform(size=(0, 1))) diff --git a/tests/python/unittest/test_numpy_op.py b/tests/python/unittest/test_numpy_op.py index 6fa22e58b918..7dd165b5421f 100644 --- a/tests/python/unittest/test_numpy_op.py +++ b/tests/python/unittest/test_numpy_op.py @@ -4164,6 +4164,7 @@ def dbg(name, data): for (iop, op) in enumerate(grad[0]): assert_almost_equal(grad[0][iop], grad[1][iop], rtol=rtol, atol=atol) + @with_seed() @use_np def test_np_diagflat():