Skip to content

Commit d04f622

Browse files
author
DvirDukhan
authored
Merge branch 'master' into ck-fixjinja-location
2 parents 4eefa63 + 6f50c27 commit d04f622

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/flow/tests_dag_errors.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,6 @@ def test_dag_error_before_tensorget_op(env):
282282
env.assertEqual(type(ret[2]), redis.exceptions.ResponseError)
283283
expected_error_msg = 'Incompatible shapes: [2] vs. [2,2,3] \t [[{{node mul}}]]'
284284
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]))
285+
expected_error_msg = 'Invalid argument: required broadcastable shapes' \
286+
' [[{{node mul}}]]'
287+
env.assertContains(expected_error_msg, str(ret[2]))

0 commit comments

Comments
 (0)