Bugfix for the required prompts for the AzureML wizard (#2959)

* fixing the required prompts * fixing the prompt text
zenml-io · Aug 28, 2024 · 990498a · 990498a
1 parent 1129bd1
commit 990498a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/zenml/cli/stack.py b/src/zenml/cli/stack.py
@@ -2477,6 +2477,12 @@ def query_region(
                 "Skypilot cluster",
                 is_skypilot=True,
             )
+        elif flavor == "azureml":
+            config["subscription_id"] = Prompt.ask(
+                "Enter the subscription ID:"
+            )
+            config["resource_group"] = Prompt.ask("Enter the resource group:")
+            config["workspace"] = Prompt.ask("Enter the workspace name:")
         elif flavor == "vertex":
             config["location"] = query_region(
                 StackDeploymentProvider.GCP, "Vertex AI job"