Update worker.py

Author: justinmerrell

runpod/serverless/worker.py

@@ -92,16 +92,18 @@ async def run_worker(config: Dict[str, Any]) -> None:
     Args:
         config (Dict[str, Any]): Configuration parameters for the worker.
     """
-    connector = aiohttp.TCPConnector(limit=None)
-    async with aiohttp.ClientSession(
-            connector=connector, headers=_get_auth_header(), timeout=_TIMEOUT) as session:
+    heartbeat.start_ping()
 
+    client_session = aiohttp.ClientSession(
+        connector=aiohttp.TCPConnector(limit=None),
+        headers=_get_auth_header(), timeout=_TIMEOUT
+    )
+
+    async with client_session as session:
         job_scaler = JobScaler(
             concurrency_controller=config.get('concurrency_controller', None)
         )
 
-        heartbeat.start_ping()
-
         while job_scaler.is_alive():
 
             async for job in job_scaler.get_jobs(session):