Fix queue_id parameter passing

- Pass queue_id to _assign_local_job method
- Fix NameError when assigning local jobs
parent d28295fc
...@@ -539,7 +539,7 @@ class ClusterMaster: ...@@ -539,7 +539,7 @@ class ClusterMaster:
# Send job assignment # Send job assignment
if client_id == "local": if client_id == "local":
# Handle local job assignment # Handle local job assignment
return await self._assign_local_job(worker_key, job_data) return await self._assign_local_job(worker_key, job_data, queue_id)
elif client_id in self.client_websockets: elif client_id in self.client_websockets:
try: try:
# Create the job assignment message # Create the job assignment message
...@@ -574,7 +574,7 @@ class ClusterMaster: ...@@ -574,7 +574,7 @@ class ClusterMaster:
return job_id return job_id
async def _assign_local_job(self, worker_key: str, job_data: dict) -> Optional[str]: async def _assign_local_job(self, worker_key: str, job_data: dict, queue_id: int = None) -> Optional[str]:
"""Assign a job to a local worker.""" """Assign a job to a local worker."""
from .models import estimate_model_vram_requirements from .models import estimate_model_vram_requirements
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment