diff --git a/src/curategpt/wrappers/clinical/clinvar_wrapper.py b/src/curategpt/wrappers/clinical/clinvar_wrapper.py index fb8fbe1..dd05ca4 100644 --- a/src/curategpt/wrappers/clinical/clinvar_wrapper.py +++ b/src/curategpt/wrappers/clinical/clinvar_wrapper.py @@ -46,7 +46,9 @@ def objects_from_dict(self, results: Dict) -> List[Dict]: obj["protein_change"] = r["protein_change"] obj["title"] = r["title"] obj["traits"] = [ - self._trait_from_dict(t) for t in r.get("trait_set", {}).get("trait", []) if isinstance(t, dict) + self._trait_from_dict(t) + for t in r.get("trait_set", {}).get("trait", []) + if isinstance(t, dict) ] objs.append(obj) return objs diff --git a/src/curategpt/wrappers/clinical/ctgov_wrapper.py b/src/curategpt/wrappers/clinical/ctgov_wrapper.py index 5c21400..f5955f3 100644 --- a/src/curategpt/wrappers/clinical/ctgov_wrapper.py +++ b/src/curategpt/wrappers/clinical/ctgov_wrapper.py @@ -1,4 +1,4 @@ -"""Chat with a KB.""" +"""Wrapper for ClinicalTrials.gov API.""" import logging import time diff --git a/src/curategpt/wrappers/clinical/hpoa_by_pub_wrapper.py b/src/curategpt/wrappers/clinical/hpoa_by_pub_wrapper.py index 0b3535a..398c4f6 100644 --- a/src/curategpt/wrappers/clinical/hpoa_by_pub_wrapper.py +++ b/src/curategpt/wrappers/clinical/hpoa_by_pub_wrapper.py @@ -1,3 +1,5 @@ +"""Wrapper for grouping Human Phenotype Ontology annotations by publication""" + from dataclasses import dataclass from typing import ClassVar diff --git a/src/curategpt/wrappers/clinical/hpoa_wrapper.py b/src/curategpt/wrappers/clinical/hpoa_wrapper.py index 9da3aa5..957afc1 100644 --- a/src/curategpt/wrappers/clinical/hpoa_wrapper.py +++ b/src/curategpt/wrappers/clinical/hpoa_wrapper.py @@ -1,4 +1,4 @@ -"""Chat with a KB.""" +"""Wrapper for Human Phenotype Ontology annotations""" import logging from copy import deepcopy diff --git a/src/curategpt/wrappers/clinical/maxoa_wrapper.py b/src/curategpt/wrappers/clinical/maxoa_wrapper.py index 72524a7..967d939 100644 --- a/src/curategpt/wrappers/clinical/maxoa_wrapper.py +++ b/src/curategpt/wrappers/clinical/maxoa_wrapper.py @@ -1,4 +1,4 @@ -"""Chat with a KB.""" +"""Wrapper for Medical Action Ontology (MAXO) annotations""" import logging from copy import deepcopy