I have added a patch which would solve this issue-- in _runInteration and _runWithConnection, added a check for the except block to determine if the exception is a connectionlost, and then raise that exception instead of the original if that is the case.