Wait for rollback to finish before ending session in transient_txn_error_labels_with_write_concern.js

Connections are closed while a node is going through rollback. This can cause race conditions where a network error can occur if a command is executed on a node while it is still going through rollback.