Add function to get publication types per PubmedArticle

indra/literature/pubmed_client.py

@@ -934,4 +934,21 @@ def get_all_ids(search_term):
     # If there are more than 10k IDs, the CLI outputs a . for each
     # iteration, these have to be filtered out
     pmids = [e for e in elements if '.' not in e]
-    return pmids
+    return pmids
+
+
+def get_publication_types(article: ET.Element):
+    """Return the set of PublicationType for the article
+
+    Parameters
+    ----------
+    article :
+        The XML element for the article. Typically, this is a PubmedArticle
+        node.
+
+    Returns
+    -------
+    : set[str]
+        A set of publication type
+    """
+    return {pt.text for pt in article.find('.//PublicationTypeList')}