nomad: refactor filter running allocs (#40)

ngine-io · Jun 26, 2023 · c434854 · c434854
1 parent 2464ba2
commit c434854
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/chaotic/cloud/nomad.py b/chaotic/cloud/nomad.py
@@ -63,8 +63,7 @@ def list_allocs(self, namespace: Optional[str] = None) -> List[dict]:
             "namespace": namespace,
         }
         r = self.query_api("get", "allocations", params=params)
-        allocs = [alloc for alloc in r.json() if alloc["ClientStatus"] == "running"]
-        return allocs
+        return r.json()
 
     def read_alloc(self, alloc_id: str) -> dict:
         r = self.query_api("get", f"allocation/{alloc_id}")
@@ -135,7 +134,7 @@ def action(self) -> None:
     def action_job(self) -> None:
         namespace = self.get_namespace()
         if namespace:
-            allocs = self.nomad.list_allocs(namespace=namespace)
+            allocs = [alloc for alloc in self.nomad.list_allocs(namespace=namespace) if alloc["ClientStatus"] == "running"]
 
             job_type_skiplist = self.configs.get("job_type_skiplist")
             if job_type_skiplist: