Skip to content

Commit 24a9ea0

Browse files
committed
Merge branch 'update_documentation' of https://github.com/RedisAI/RedisAI into update_documentation
2 parents e7b63b8 + 9554bc8 commit 24a9ea0

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)