Adjust update_doc_array snippet to not require an extra import (GoogleCloudPlatform#2306)

jlara310 · BenWhitehead · commit c9337cdfa694 · 2019-07-30T11:12:37.000-04:00
diff --git a/firestore/cloud-client/snippets.py b/firestore/cloud-client/snippets.py
@@ -15,7 +15,6 @@
 from time import sleep
 
 from google.cloud import firestore
-from google.cloud.firestore_v1 import ArrayRemove, ArrayUnion
 import google.cloud.exceptions
 
 
@@ -319,10 +318,10 @@ def update_doc_array():
     city_ref = db.collection(u'cities').document(u'DC')
 
     # Atomically add a new region to the 'regions' array field.
-    city_ref.update({u'regions': ArrayUnion([u'greater_virginia'])})
+    city_ref.update({u'regions': firestore.ArrayUnion([u'greater_virginia'])})
 
     # // Atomically remove a region from the 'regions' array field.
-    city_ref.update({u'regions': ArrayRemove([u'east_coast'])})
+    city_ref.update({u'regions': firestore.ArrayRemove([u'east_coast'])})
     # [END fs_update_doc_array]
     city = city_ref.get()
     print(u'Updated the regions field of the DC. {}'.format(city.to_dict()))
