Adding _grpc_catch_rendezvous to _DatastoreAPIOverGRPC.lookup.

Missed in googleapis#2590.
richkadel · Oct 25, 2016 · 91437f4 · 91437f4
1 parent 842c486
commit 91437f4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/datastore/google/cloud/datastore/connection.py b/datastore/google/cloud/datastore/connection.py
@@ -300,7 +300,8 @@ def lookup(self, project, request_pb):
         :returns: The returned protobuf response object.
         """
         request_pb.project_id = project
-        return self._stub.Lookup(request_pb)
+        with _grpc_catch_rendezvous():
+            return self._stub.Lookup(request_pb)
 
     def run_query(self, project, request_pb):
         """Perform a ``runQuery`` request.