Fix bug

replit_river/session.py

@@ -509,7 +509,6 @@ async def close(self, is_unexpected_close: bool) -> None:
             await self._task_manager.cancel_all_tasks()
             # TODO: unexpected_close should close stream differently here to
             # throw exception correctly.
-            logging.error(f"##### closing all streams here : {self._streams.values()}")
             for stream in self._streams.values():
                 stream.close()
             async with self._stream_lock:

replit_river/transport.py

@@ -34,7 +34,8 @@ async def close_all_sessions(self) -> None:
             f"start closing transport {self._transport_id}, number sessions : "
             f"{len(sessions)}"
         )
-        for session in sessions:
+        sessions_to_close = list(sessions)
+        for session in sessions_to_close:
             await session.close(False)
         logging.info(f"Transport closed {self._transport_id}")