Throttle tunnel connection attempts

nerfstudio-project · Sep 16, 2023 · ca39637 · ca39637
1 parent dde0373
commit ca39637
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/viser/_tunnel.py b/src/viser/_tunnel.py
@@ -135,14 +135,17 @@ async def connect(
                 ],
                 return_when=asyncio.FIRST_COMPLETED,
             )
-        except Exception:
+        except Exception as e:
             pass
         finally:
             if local_w is not None:
                 local_w.close()
             if remote_w is not None:
                 remote_w.close()
 
+        # Throttle connection attempts.
+        await asyncio.sleep(0.1)
+
 
 if __name__ == "__main__":
     tunnel = _ViserTunnel(8080)