It looks like the driver does not currently add the UnknownTransactionCommitResult or TransientTransactionError labels to SocketTimeoutErrors that occur during transactions. We should add integration tests for these cases and fix the behavior so the driver properly retries in those cases.