further refactor _create_document

NatLibFi · Aug 9, 2019 · 0678998 · 0678998
1 parent 4acc731
commit 0678998
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/annif/corpus/types.py b/annif/corpus/types.py
@@ -29,10 +29,11 @@ def _create_document(self, text, uris, labels):
         information. URIs for labels and vice versa are looked up from the
         subject index, if available."""
 
-        if not uris and labels and self._subject_index:
-            uris = set((self._subject_index.labels_to_uris(labels)))
-        if not labels and uris and self._subject_index:
-            labels = set((self._subject_index.uris_to_labels(uris)))
+        if self._subject_index:
+            if not uris and labels:
+                uris = set((self._subject_index.labels_to_uris(labels)))
+            if not labels and uris:
+                labels = set((self._subject_index.uris_to_labels(uris)))
 
         return Document(text=text, uris=uris, labels=labels)