diff --git a/django_elasticsearch_dsl/registries.py b/django_elasticsearch_dsl/registries.py index 66690486..569442cb 100644 --- a/django_elasticsearch_dsl/registries.py +++ b/django_elasticsearch_dsl/registries.py @@ -152,9 +152,9 @@ def get_documents(self, models=None): Get all documents in the registry or the documents for a list of models """ if models is not None: - return set(chain(*(self._models[model] for model in models - if model in self._models))) - return set(chain(*itervalues(self._indices))) + return set(chain.from_iterable(self._models[model] for model in models + if model in self._models)) + return set(chain.from_iterable(itervalues(self._indices))) def get_models(self): """