We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents e7b63b8 + 9554bc8 commit 24a9ea0Copy full SHA for 24a9ea0
1 file changed
tests/flow/tests_dag_errors.py
@@ -282,7 +282,6 @@ def test_dag_error_before_tensorget_op(env):
282
env.assertEqual(type(ret[2]), redis.exceptions.ResponseError)
283
expected_error_msg = 'Incompatible shapes: [2] vs. [2,2,3] \t [[{{node mul}}]]'
284
if DEVICE != 'CPU':
285
- expected_error_msg = '2 root error(s) found. (0) Invalid argument: required broadcastable shapes' \
286
- ' [[{{node mul}}]] (1) Invalid argument: required broadcastable shapes' \
287
- ' [[{{node mul}}]] [[mul/_5]] 0 successful operations. 0 derived errors ignored.'
288
- env.assertEqual(expected_error_msg, str(ret[2]))
+ expected_error_msg = 'Invalid argument: required broadcastable shapes' \
+ ' [[{{node mul}}]]'
+ env.assertContains(expected_error_msg, str(ret[2]))
0 commit comments