[BACKPORT 2.20.1] [PLAT-12124] Fix rolling restart to be be retriable.

Summary: This issue was caused due to an incorrect merge resolved on 2.20 branch.

Test Plan:
Testing done: {F140873}

Build and test code on 2.20.1

Reviewers: sanketh, yugaware, nsingh

Reviewed By: sanketh

Subscribers: yugaware

Differential Revision: https://phorge.dev.yugabyte.com/D31525

managed/src/main/java/com/yugabyte/yw/common/CustomerTaskManager.java

@@ -24,6 +24,7 @@
 import com.yugabyte.yw.forms.KubernetesGFlagsUpgradeParams;
 import com.yugabyte.yw.forms.KubernetesOverridesUpgradeParams;
 import com.yugabyte.yw.forms.ResizeNodeParams;
+import com.yugabyte.yw.forms.RestartTaskParams;
 import com.yugabyte.yw.forms.RestoreBackupParams;
 import com.yugabyte.yw.forms.SoftwareUpgradeParams;
 import com.yugabyte.yw.forms.UniverseDefinitionTaskParams;
@@ -503,6 +504,9 @@ public CustomerTask retryCustomerTask(UUID customerUUID, UUID taskUUID) {
       case CloudProviderDelete:
         taskParams = Json.fromJson(oldTaskParams, CloudProviderDelete.Params.class);
         break;
+      case RestartUniverseKubernetesUpgrade:
+        taskParams = Json.fromJson(oldTaskParams, RestartTaskParams.class);
+        break;
       default:
         String errMsg =
             String.format(