File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed
Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff 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 ]))
You can’t perform that action at this time.
0 commit comments