Skip to content

Commit 86b4ce7

Browse files
authored
Use chain.from_iterable in registries.py (#278)
1 parent 805d391 commit 86b4ce7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

django_elasticsearch_dsl/registries.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -152,9 +152,9 @@ def get_documents(self, models=None):
152152
Get all documents in the registry or the documents for a list of models
153153
"""
154154
if models is not None:
155-
return set(chain(*(self._models[model] for model in models
156-
if model in self._models)))
157-
return set(chain(*itervalues(self._indices)))
155+
return set(chain.from_iterable(self._models[model] for model in models
156+
if model in self._models))
157+
return set(chain.from_iterable(itervalues(self._indices)))
158158

159159
def get_models(self):
160160
"""

0 commit comments

Comments
 (0)