As identified in https://github.com/data-apis/array-api-tests/pull/101#issuecomment-1949511287 we'll need to update `test_sum`, `test_prod` and `test_trace` because of https://github.com/data-apis/array-api/issues/731