fix test

Signed-off-by: hongkuanz <[email protected]>

Author: tedzhouhk

tests/profiler/test_profile_sla_aiconfigurator.py

@@ -66,6 +66,7 @@ def __init__(self):
                 self.aic_backend_version = None
                 self.num_gpus_per_node = 8
                 self.deploy_after_profile = False
+                self.pick_with_webui = False
                 # Provide minimal model_info to avoid HF queries
                 self.model_info = ModelInfo(
                     model_size=16384.0,

tests/profiler/test_profile_sla_dryrun.py

@@ -73,6 +73,7 @@ def __init__(self):
                 self.aic_backend_version = None
                 self.num_gpus_per_node = 8
                 self.deploy_after_profile = False
+                self.pick_with_webui = False
                 # Provide minimal model_info to avoid HF queries
                 self.model_info = ModelInfo(
                     model_size=16384.0,
@@ -116,6 +117,7 @@ def __init__(self):
                 self.aic_backend_version = None
                 self.num_gpus_per_node = 8
                 self.deploy_after_profile = False
+                self.pick_with_webui = False
                 self.model_info = ModelInfo(
                     model_size=16384.0,
                     architecture="TestArchitecture",
@@ -180,6 +182,7 @@ def __init__(self):
                 self.aic_backend_version = None
                 self.num_gpus_per_node = 8
                 self.deploy_after_profile = False
+                self.pick_with_webui = False
                 self.model_info = ModelInfo(
                     model_size=16384.0,
                     architecture="TestArchitecture",
@@ -233,6 +236,7 @@ def __init__(self):
                 self.aic_backend_version = None
                 self.num_gpus_per_node = 8
                 self.deploy_after_profile = False
+                self.pick_with_webui = False
                 self.model_info = ModelInfo(
                     model_size=65536.0,
                     architecture="TestMoEArchitecture",
@@ -309,6 +313,7 @@ def __init__(self):
                 # Set to 0 to trigger auto-generation path
                 self.num_gpus_per_node = 0
                 self.deploy_after_profile = False
+                self.pick_with_webui = False
                 self.enable_gpu_discovery = True
 
         return Args()
@@ -376,6 +381,7 @@ def __init__(self):
                 self.aic_backend_version = None
                 self.num_gpus_per_node = 0
                 self.deploy_after_profile = False
+                self.pick_with_webui = False
                 self.enable_gpu_discovery = True
 
         return Args()
@@ -443,6 +449,7 @@ def __init__(self):
                 self.aic_backend_version = None
                 self.num_gpus_per_node = 0
                 self.deploy_after_profile = False
+                self.pick_with_webui = False
                 self.enable_gpu_discovery = True
 
         return Args()