Fixed googleapis#86 - Entity objects from a fetched query have a prop…

…er Dataset reference.
jgeewax · Apr 22, 2014 · 81f109e · 81f109e
1 parent 1431fa2
commit 81f109e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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]