I'm running code from Colab and my DB connection keeps having issues. is there a way to prevent it dropping in the middle of 4+ min long operations?