FIX AutoDL调度修复暖池功能
This commit is contained in:
@@ -53,20 +53,17 @@ class Server:
|
||||
return {"status": "queuing", "msg":""}
|
||||
|
||||
def introspect_instance(self):
|
||||
loguru.logger.info("Introspecting worker started")
|
||||
loguru.logger.info("Introspecting worker started || Scaledown Window: %ds" % self.instance_pool.scaledown_window)
|
||||
while True:
|
||||
self.instance_pool.introspection()
|
||||
time.sleep(0.5)
|
||||
time.sleep(1)
|
||||
|
||||
def scaling_worker(self):
|
||||
loguru.logger.info("Scaling worker started")
|
||||
try:
|
||||
last_target = 0
|
||||
while True:
|
||||
# 提交任务
|
||||
if last_target != self.waiting_queue.get_size()+self.running_pool.get_running_size():
|
||||
last_target = self.waiting_queue.get_size()+self.running_pool.get_running_size()
|
||||
self.instance_pool.scale_instance(last_target)
|
||||
self.instance_pool.scale_instance(self.waiting_queue.get_size()+self.running_pool.get_running_size(), disable_shrink=True)
|
||||
for instance in self.instance_pool.instances:
|
||||
if not instance.active:
|
||||
# 从等待队列取出任务
|
||||
|
||||
Reference in New Issue
Block a user