make PipelineRunner._gc_collect async

This commit is contained in:
Om Chauhan
2026-04-08 09:27:18 +05:30
parent 0acfb4dd49
commit 225ea907d5

View File

@@ -90,7 +90,7 @@ class PipelineRunner(BaseObject):
await self._sig_task await self._sig_task
if self._force_gc: if self._force_gc:
self._gc_collect() await self._gc_collect()
logger.debug(f"Runner {self} finished running {task}") logger.debug(f"Runner {self} finished running {task}")
@@ -136,8 +136,8 @@ class PipelineRunner(BaseObject):
logger.warning(f"Interruption detected. Cancelling runner {self}") logger.warning(f"Interruption detected. Cancelling runner {self}")
await self.cancel() await self.cancel()
def _gc_collect(self): async def _gc_collect(self):
"""Force garbage collection and log results.""" """Force garbage collection and log results."""
collected = gc.collect() collected = await asyncio.to_thread(gc.collect)
logger.debug(f"Garbage collector: collected {collected} objects.") logger.debug(f"Garbage collector: collected {collected} objects.")
logger.debug(f"Garbage collector: uncollectable objects {gc.garbage}") logger.debug(f"Garbage collector: uncollectable objects {gc.garbage}")