Hohoho / app /controllers /manager /memory_manager.py
Arnav Singh
Create English Hugging Face Docker Space edition
cc2bdee
Raw History Blame Contribute Delete
479 Bytes
from queue import Queue
from typing import Dict
from app.controllers.manager.base_manager import TaskManager
class InMemoryTaskManager(TaskManager):
def create_queue(self):
return Queue(maxsize=self.max_queued_tasks)
def enqueue(self, task: Dict):
self.queue.put(task)
def dequeue(self):
return self.queue.get()
def is_queue_empty(self):
return self.queue.empty()
def queue_size(self):
return self.queue.qsize()