From 81f109ea1cad242badbf4e4db8a6073b77eddb7a Mon Sep 17 00:00:00 2001 From: JJ Geewax <jj@geewax.org> Date: Tue, 22 Apr 2014 12:52:13 -0400 Subject: [PATCH] Fixed #86 - Entity objects from a fetched query have a proper Dataset reference. --- gcloud/datastore/query.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/gcloud/datastore/query.py b/gcloud/datastore/query.py index 0e209b01e12b..e8838de24951 100644 --- a/gcloud/datastore/query.py +++ b/gcloud/datastore/query.py @@ -311,4 +311,5 @@ def fetch(self, limit=None): entity_pbs = self.dataset().connection().run_query( query_pb=clone.to_protobuf(), dataset_id=self.dataset().id()) - return [Entity.from_protobuf(entity) for entity in entity_pbs] + return [Entity.from_protobuf(entity, dataset=self.dataset()) + for entity in entity_pbs]