Skip to content

Commit

Permalink
More linting and docstrings for wrappers
Browse files Browse the repository at this point in the history
  • Loading branch information
caufieldjh committed Nov 15, 2024
1 parent bc586d8 commit 5ea8b1c
Show file tree
Hide file tree
Showing 5 changed files with 8 additions and 4 deletions.
4 changes: 3 additions & 1 deletion src/curategpt/wrappers/clinical/clinvar_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion src/curategpt/wrappers/clinical/ctgov_wrapper.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
"""Chat with a KB."""
"""Wrapper for ClinicalTrials.gov API."""

import logging
import time
Expand Down
2 changes: 2 additions & 0 deletions src/curategpt/wrappers/clinical/hpoa_by_pub_wrapper.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
"""Wrapper for grouping Human Phenotype Ontology annotations by publication"""

from dataclasses import dataclass
from typing import ClassVar

Expand Down
2 changes: 1 addition & 1 deletion src/curategpt/wrappers/clinical/hpoa_wrapper.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
"""Chat with a KB."""
"""Wrapper for Human Phenotype Ontology annotations"""

import logging
from copy import deepcopy
Expand Down
2 changes: 1 addition & 1 deletion src/curategpt/wrappers/clinical/maxoa_wrapper.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
"""Chat with a KB."""
"""Wrapper for Medical Action Ontology (MAXO) annotations"""

import logging
from copy import deepcopy
Expand Down

0 comments on commit 5ea8b1c

Please sign in to comment.