Merge pull request #87 from jgeewax/86-fix-dataset-reference

Fix #86 - Entity objects from a query have a proper Dataset reference
googleapis · Apr 22, 2014 · 9654c3b · 9654c3b
2 parents 4b56a23 + 86b886e
commit 9654c3b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gcloud/datastore/query.py b/gcloud/datastore/query.py
@@ -244,4 +244,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]