update comments

Signed-off-by: pandyamarut <pandyamarut@gmail.com>

Author: pandyamarut

runpod/serverless/modules/rp_scale.py

@@ -176,6 +176,15 @@ def start(self):
         asyncio.run(self.run())
 
     def handle_shutdown(self, signum, frame):
+        """
+        Called when the worker is signalled to shut down.
+        This function is called when the worker receives a signal to shut down, such as
+        SIGTERM or SIGINT. It sets the shutdown event, which will cause the worker to
+        exit its main loop and shut down gracefully.
+        Args:
+            signum: The signal number that was received.
+            frame: The current stack frame.
+        """
         log.debug(f"Received shutdown signal: {signum}.")
         self.kill_worker()
 
@@ -197,9 +206,15 @@ async def run(self):
                 raise
 
     def is_alive(self):
+        """
+        Return whether the worker is alive or not.
+        """
         return not self._shutdown_event.is_set()
 
     def kill_worker(self):
+        """
+        Whether to kill the worker.
+        """
         log.debug("Kill worker.")
         self._shutdown_event.set()
 

runpod/serverless/modules/worker_state.py

@@ -58,7 +58,7 @@ def __str__(self) -> str:
 
 class JobsProgress:
     """
-    OPTIMIZED: Track jobs in progress with O(1) operations using threading.Lock
+    Track jobs in progress with min operations using threading.Lock
     instead of multiprocessing.Manager for better performance.
     """
 
@@ -91,7 +91,7 @@ def clear(self) -> None:
 
     def add(self, element: Any):
         """
-        OPTIMIZED: O(1) addition of jobs using dict
+        addition of jobs using dict
         """
         if isinstance(element, str):
             job_id = element