rebase updates

aws · Jul 12, 2024 · 4fc2e3c · 4fc2e3c
1 parent 072f132
commit 4fc2e3c
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/test/suites/scale/deprovisioning_test.go b/test/suites/scale/deprovisioning_test.go
@@ -101,11 +101,9 @@ var _ = Describe("Deprovisioning", Label(debug.NoWatch), Label(debug.NoEvents),
 				},
 			},
 		}...)
-		nodePool.Spec.Disruption.Budgets = []corev1beta1.Budget{
-			{
-				Nodes: "70%",
-			},
-		}
+		nodePool.Spec.Disruption.Budgets = []karpv1.Budget{{
+			Nodes: "70%",
+		}}
 		deploymentOptions = test.DeploymentOptions{
 			PodOptions: test.PodOptions{
 				ResourceRequirements: corev1.ResourceRequirements{
@@ -302,11 +300,9 @@ var _ = Describe("Deprovisioning", Label(debug.NoWatch), Label(debug.NoEvents),
 			env.MeasureDeprovisioningDurationFor(func() {
 				By("enabling deprovisioning across nodePools")
 				for _, p := range nodePoolMap {
-					p.Spec.Disruption.Budgets = []corev1beta1.Budget{
-						{
-							Nodes: "70%",
-						},
-					}
+					p.Spec.Disruption.Budgets = []karpv1.Budget{{
+						Nodes: "70%",
+					}}
 					env.ExpectCreatedOrUpdated(p)
 				}
 				env.ExpectUpdated(driftNodeClass)