set edition to ENTERPRISE in reservation_create

dizcology · dandhlee · commit 73fa77c13f6b · 2023-07-24T10:30:23.000-04:00
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,
