From 2b7712dc9f512c00b12edf17f2b26e64646fc09e Mon Sep 17 00:00:00 2001 From: David Hall Date: Fri, 11 Oct 2024 14:21:38 -0700 Subject: [PATCH] shut up about scaling down --- src/levanter/utils/actor_pool.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/levanter/utils/actor_pool.py b/src/levanter/utils/actor_pool.py index f40834bb5..a694bee20 100644 --- a/src/levanter/utils/actor_pool.py +++ b/src/levanter/utils/actor_pool.py @@ -133,7 +133,7 @@ def _adjust_pool_size(self): # Schedule scale down if idle elif num_pending_tasks == 0 and num_nonworking_actors > self._min_size: - if self._scale_down_task is None or self._scale_down_task.done(): + if self._scale_down_task is None: self._scale_down_task = asyncio.create_task(self._schedule_scale_down()) async def _schedule_scale_down(self): @@ -142,8 +142,9 @@ async def _schedule_scale_down(self): if self.num_pending_tasks == 0: logger.info("Scaling down due to no pending tasks.") self._scale_down(self._min_size) + self._scale_down_task = None except asyncio.CancelledError: - logger.info("Scale down task was cancelled due to new activity.") + logger.debug("Scale down task was cancelled due to new activity.") def _get_object_location(self, obj_ref: ray.ObjectRef) -> Optional[str]: """Get the location of the given object reference."""