set edition to ENTERPRISE in reservation_create

GoogleCloudPlatform · Jul 24, 2023 · 73fa77c · 73fa77c
1 parent 163d964
commit 73fa77c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/bigquery-reservation/snippets/reservation_create.py b/bigquery-reservation/snippets/reservation_create.py
@@ -67,7 +67,9 @@ def create_reservation(
 
     parent = reservation_client.common_location_path(project_id, location)
 
-    reservation = reservation_types.Reservation(slot_capacity=slot_capacity)
+    reservation = reservation_types.Reservation(
+        slot_capacity=slot_capacity, edition=reservation_types.Edition.ENTERPRISE
+    )
     reservation = reservation_client.create_reservation(
         parent=parent,
         reservation=reservation,