From 634524f4e554c4405b3917dafcb8046b9fc4a80e Mon Sep 17 00:00:00 2001 From: Meraj Date: Thu, 25 Mar 2021 19:47:50 -0400 Subject: [PATCH] Remove `test_trace_raises` for dims more than 25 --- tensornetwork/backends/pytorch/pytorch_backend_test.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/tensornetwork/backends/pytorch/pytorch_backend_test.py b/tensornetwork/backends/pytorch/pytorch_backend_test.py index 7fcc5f256..cbedf9952 100644 --- a/tensornetwork/backends/pytorch/pytorch_backend_test.py +++ b/tensornetwork/backends/pytorch/pytorch_backend_test.py @@ -630,14 +630,6 @@ def test_trace(dtype, offset, axis1, axis2): np.testing.assert_allclose(actual, expected, atol=1e-6, rtol=1e-6) -def test_trace_raises(): - shape = tuple([1] * 30) - backend = pytorch_backend.PyTorchBackend() - array = backend.randn(shape, seed=10) - with pytest.raises(ValueError): - _ = backend.trace(array) - - @pytest.mark.parametrize("pivot_axis", [-1, 1, 2]) @pytest.mark.parametrize("dtype", torch_randn_dtypes) def test_pivot(dtype, pivot_axis):