Browse Source

fix occasional fail of ut of tensor processor.

tags/v1.1.0
wangshuide2020 5 years ago
parent
commit
9cffadb212
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tests/ut/datavisual/processors/test_tensor_processor.py

+ 1
- 1
tests/ut/datavisual/processors/test_tensor_processor.py View File

@@ -185,7 +185,7 @@ class TestTensorProcessor:
# Compare tensor shape when recv_tensor shape is not empty.
if recv_tensor.shape != (0,):
assert recv_tensor.shape == expected_tensor.shape
assert np.sum(np.isclose(recv_tensor, expected_tensor, rtol=1e-6) == 0) == 0
assert np.allclose(recv_tensor, expected_tensor, rtol=1e-6)

@pytest.mark.usefixtures('load_tensor_record')
def test_get_tensor_stats_success(self):


Loading…
Cancel
Save