From 225c159f6cdde99f4029544a6f6a67395bd0c887 Mon Sep 17 00:00:00 2001 From: Jasmin Makwana Date: Thu, 14 May 2026 13:00:53 +0530 Subject: [PATCH] [patch] Add option to skip scheduling configuration for AI Service tenant --- python/src/mas/cli/aiservice/install/app.py | 4 +++- python/src/mas/cli/install/app.py | 4 +++- python/test/aiservice/install/test_app.py | 2 ++ 3 files changed, 8 insertions(+), 2 deletions(-) diff --git a/python/src/mas/cli/aiservice/install/app.py b/python/src/mas/cli/aiservice/install/app.py index 81c8707f96b..2e862801f78 100644 --- a/python/src/mas/cli/aiservice/install/app.py +++ b/python/src/mas/cli/aiservice/install/app.py @@ -720,7 +720,9 @@ def configSchedulingConstraints(self): " `nodeSelector`: Dictionary of node label key-value pairs", ]) - self.aiserviceTenantSchedulingConfigFileLocal = self.promptForFile("Scheduling configuration YAML file", mustExist=True, envVar="AISERVICE_TENANT_SCHEDULING_CONFIG_FILE") + configSchedulingConstraints = self.yesOrNo('Configure Scheduling policies for AI Service tenant') + if configSchedulingConstraints: + self.aiserviceTenantSchedulingConfigFileLocal = self.promptForFile("Scheduling configuration YAML file", mustExist=True, envVar="AISERVICE_TENANT_SCHEDULING_CONFIG_FILE") def _setMinioStorageDefaults(self) -> None: """ diff --git a/python/src/mas/cli/install/app.py b/python/src/mas/cli/install/app.py index 54daf37cca5..8b098fab5ab 100644 --- a/python/src/mas/cli/install/app.py +++ b/python/src/mas/cli/install/app.py @@ -1562,7 +1562,9 @@ def configSchedulingConstraints(self): " `nodeSelector`: Dictionary of node label key-value pairs", ]) - self.aiserviceTenantSchedulingConfigFileLocal = self.promptForFile("Scheduling configuration YAML file", mustExist=True, envVar="AISERVICE_TENANT_SCHEDULING_CONFIG_FILE") + configSchedulingConstraints = self.yesOrNo('Configure Scheduling policies for AI Service tenant') + if configSchedulingConstraints: + self.aiserviceTenantSchedulingConfigFileLocal = self.promptForFile("Scheduling configuration YAML file", mustExist=True, envVar="AISERVICE_TENANT_SCHEDULING_CONFIG_FILE") @logMethodCall def _setMinioStorageDefaults(self) -> None: diff --git a/python/test/aiservice/install/test_app.py b/python/test/aiservice/install/test_app.py index f9c7f0b9aa9..dc20292abfc 100644 --- a/python/test/aiservice/install/test_app.py +++ b/python/test/aiservice/install/test_app.py @@ -180,6 +180,8 @@ def set_mixin_prompt_input(**kwargs): return '' if re.match('.*Instance ID.*', message): return 'apmdevops' + if re.match('.*Configure Scheduling policies for AI Service tenant.*', message): + return 'y' if re.match('.*Scheduling constraints YAML file.*', message): return f'{tmpdir}/aiservice-tenant-affinity-config.yaml' if re.match('.*Operational Mode.*', message):