hotfix load distribution

Author: ChristianHinge

nodes/manager/manager.py

@@ -232,10 +232,12 @@ def get_addr_to_run_node(self, node_name):
         # return node_name
 
         if node_name in self.nodes.keys():
-            return "localhost:8000"
+            lowest_load = self.queue.get_load()
+            addr_with_lowest_load = "localhost:8000"
+        else:
+            lowest_load = None
+            addr_with_lowest_load = None
 
-        lowest_load = None
-        addr_with_lowest_load = None
         # query other hosts for available gpu
         for addr in self.other_addrs:
             url = f"http://{addr}/manager/dispatcher/has_node/{node_name}"

rhnode/version.py

@@ -1,4 +1,4 @@
-__version__ = "1.2.0"
+__version__ = "1.3.0-a.1"
 git_revision = ""
 
 

tests/docker-compose.yaml

@@ -88,3 +88,19 @@ services:
     environment:
       #RH_EMAIL_ON_ERROR: claes.noehr.ladefoged@regionh.dk
       TZ: "Europe/Copenhagen"
+
+  totalsegmentator:
+      shm_size: '24gb'
+      image: rhnode/totalsegmentator:v1.0.0_rhnode1.2.0
+      deploy:
+        resources:
+          reservations:
+            devices:
+              - driver: nvidia
+                count: 1
+                capabilities: [gpu]
+      expose:
+        - "8000"
+        
+      labels:
+        - "traefik.http.routers.totalsegmentator.rule=PathPrefix(`/totalsegmentator`)"