Use chain.from_iterable in registries.py (#278)

django-es · Jun 18, 2020 · 86b4ce7 · 86b4ce7
1 parent 805d391
commit 86b4ce7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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):
         """