This results in an error:

mutation {
  update_smth_by_pk(pk_columns: { id: 1 }, _set: ...) {
    id
    rel {
      id
    }
  }
}

Update: the errors appears only for update_by_pk mutations,
update(where..) { returning { ... } } works