From 2b5260e395dace43d3d342c3a268b6ebe5badd2e Mon Sep 17 00:00:00 2001
From: smokestacklightnin
 <125844868+smokestacklightnin@users.noreply.github.com>
Date: Sun, 10 Nov 2024 00:02:40 -0800
Subject: [PATCH] Add `attempt_fixes` to
 `Upgrade_2024_11_1._version_specific_upgrade`

---
 src/_nebari/upgrade.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/_nebari/upgrade.py b/src/_nebari/upgrade.py
index 41e27c49c6..3012a5b7cf 100644
--- a/src/_nebari/upgrade.py
+++ b/src/_nebari/upgrade.py
@@ -1291,7 +1291,13 @@ class Upgrade_2024_11_1(UpgradeStep):
 
     @override
     def _version_specific_upgrade(
-        self, config, start_version, config_filename: Path, *args, **kwargs
+        self,
+        config,
+        start_version,
+        config_filename: Path,
+        *args,
+        attempt_fixes=False,
+        **kwargs,
     ):
         if config.get("provider", "") == ProviderEnum.azure.value:
             rich.print("\n ⚠️ Upgrade Warning ⚠️")
@@ -1342,12 +1348,11 @@ def _version_specific_upgrade(
 
         # Prompt the user for role assignment (if yes, transforms the response into bool)
         assign_roles = (
-            Prompt.ask(
+            attempt_fixes
+            or Confirm.ask(
                 "[bold]Would you like Nebari to assign the corresponding role/scopes to all of your current groups automatically?[/bold]",
-                choices=["y", "N"],
-                default="N",
+                default=False,
             ).lower()
-            == "y"
         )
 
         if assign_roles: