Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions gcloud/datastore/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
from gcloud.datastore.batch import Batch
from gcloud.datastore.entity import Entity
from gcloud.datastore.key import Key
from gcloud.datastore.key import _dataset_ids_equal
from gcloud.datastore.query import Query
from gcloud.datastore.transaction import Transaction
from gcloud.environment_vars import DATASET
Expand Down Expand Up @@ -288,9 +289,10 @@ def get_multi(self, keys, missing=None, deferred=None):
if not keys:
return []

ids = list(set([key.dataset_id for key in keys]))
if ids != [self.dataset_id]:
raise ValueError('Keys do not match dataset ID')
ids = set(key.dataset_id for key in keys)
for ds_id in ids:

This comment was marked as spam.

This comment was marked as spam.

This comment was marked as spam.

if not _dataset_ids_equal(ds_id, self.dataset_id):
raise ValueError('Keys do not match dataset ID')

transaction = self.current_transaction

Expand Down