Browse Source

!779 fix occasional fail of ut of tensor processor

Merge pull request !779 from wangshuide/wsd1010
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
768f4dc838
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