sros: fix distributed model (getMem, getCpu) (#199)

hellt · Jun 12, 2024 · 65e21e3 · 65e21e3
1 parent 249b3ca
commit 65e21e3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/sros/docker/launch.py b/sros/docker/launch.py
@@ -1229,8 +1229,8 @@ def __init__(self, hostname, username, password, mode, variant, conn_mode):
         # cp - control plane. role is used to create a separate overlay image name
         self.role = "cp"
 
-        ram: int = vrnetlab.getMem(self.role, variant.get("cp").get("min_ram"))
-        cpu: int = vrnetlab.getCpu(self.role, variant.get("cp").get("cpu"))
+        ram: int = getMem(self.role, variant.get("cp").get("min_ram"))
+        cpu: int = getCpu(self.role, variant.get("cp").get("cpu"))
         slot: str = variant.get("cp").get("slot")
 
         super(SROS_cp, self).__init__(
@@ -1303,8 +1303,8 @@ def __init__(self, lc_config, conn_mode, num_nics, slot=1, nic_eth_start=1):
         # role lc if for a line card. role is used to create a separate overlay image name
         self.role = "lc"
 
-        ram: int = vrnetlab.getMem(self.role, lc_config.get("min_ram"))
-        cpu: int = vrnetlab.getCpu(self.role, lc_config.get("cpu"))
+        ram: int = getMem(self.role, lc_config.get("min_ram"))
+        cpu: int = getCpu(self.role, lc_config.get("cpu"))
 
         super(SROS_lc, self).__init__(
             None,