From 3e17e79df57daa8571f3e2073d46656c4f8e5030 Mon Sep 17 00:00:00 2001 From: croy Date: Mon, 3 Mar 2025 16:01:34 -0500 Subject: [PATCH] Fixing definition of compute Change-Id: Ie0d22676041eea8034fae65a7d39cacde125f123 --- config/lab/objects/lab_config.py | 2 +- scripts/lab_capability_scanner.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/config/lab/objects/lab_config.py b/config/lab/objects/lab_config.py index 2670f8c3..6eb3efa7 100644 --- a/config/lab/objects/lab_config.py +++ b/config/lab/objects/lab_config.py @@ -213,7 +213,7 @@ class LabConfig: """ nodes = self.get_nodes() - computes = [node for node in nodes if node.node_type == 'Compute'] + computes = [node for node in nodes if node.node_type == 'worker'] return computes def get_compute(self, compute_name) -> Optional[Node]: diff --git a/scripts/lab_capability_scanner.py b/scripts/lab_capability_scanner.py index 9b5281e6..ca317344 100644 --- a/scripts/lab_capability_scanner.py +++ b/scripts/lab_capability_scanner.py @@ -355,6 +355,10 @@ def scan_hosts(lab_config: LabConfig, ssh_connection: SSHConnection) -> [Node]: node.append_node_capability('lab_has_sriov') lab_config.add_lab_capability('lab_has_sriov') + if node.get_type() == 'worker': + node.append_node_capability('lab_has_compute') + lab_config.add_lab_capability('lab_has_compute') + if node.get_type() == 'worker' or 'worker' in node.get_subfunctions(): node.append_node_capability('lab_has_worker') lab_config.add_lab_capability('lab_has_worker')