Browse Source

default to fifo

justheuristic 2 years ago
parent
commit
b7ed72c4d6
2 changed files with 3 additions and 2 deletions
  1. 2 1
      src/server/backend.py
  2. 1 1
      src/server/task_prioritizer.py

+ 2 - 1
src/server/backend.py

@@ -1,11 +1,12 @@
 """Code for serving bloom blocks via hivemind-server"""
+import ctypes
 import multiprocessing as mp
 import os
 import threading
 from concurrent.futures import Future
 from dataclasses import dataclass, field
 from queue import Empty, PriorityQueue
-from typing import Any, Dict, Optional, Sequence, Tuple
+from typing import Any, Dict, Optional, Sequence, Tuple, List
 
 import torch
 from hivemind import BatchTensorDescriptor, use_hivemind_log_handler

+ 1 - 1
src/server/task_prioritizer.py

@@ -16,5 +16,5 @@ class TaskPrioritizerBase(ABC):
 class DummyTaskPrioritizer(TaskPrioritizerBase):
     """Simple implementation of DustBroker which counts amount of dust per task size"""
 
-    def __call__(self, *input: torch.Tensor, points: float = 0.0, **kwargs) -> float:
+    def prioritize(self, *input: torch.Tensor, points: float = 0.0, **kwargs) -> float:
         return 0.0