From e444ee4b3bffc0a979d3593ed442a2854622cb63 Mon Sep 17 00:00:00 2001
From: Martin Larralde <martin.larralde@embl.de>
Date: Mon, 7 Oct 2024 22:18:13 +0200
Subject: [PATCH] Add type annotations for `TopHits.query` property
---
pyhmmer/hmmer.py | 8 ++++----
pyhmmer/plan7.pyi | 2 ++
2 files changed, 6 insertions(+), 4 deletions(-)
diff --git a/pyhmmer/hmmer.py b/pyhmmer/hmmer.py
index 685dc9b5..4b5a9bd7 100644
--- a/pyhmmer/hmmer.py
+++ b/pyhmmer/hmmer.py
@@ -1533,8 +1533,8 @@ if __name__ == "__main__":
print(
hit.name.decode(),
(hit.accession or b"-").decode(),
- (hits.query_name or b"-").decode(),
- (hits.query_accession or b"-").decode(),
+ (hits.query.name or b"-").decode(),
+ (hits.query.accession or b"-").decode(),
hit.evalue,
hit.score,
hit.bias,
@@ -1655,8 +1655,8 @@ if __name__ == "__main__":
print(
hit.name.decode(),
(hit.accession or b"-").decode(),
- (hits.query_name or b"-").decode(),
- (hits.query_accession or b"-").decode(),
+ (hits.query.name or b"-").decode(),
+ (hits.query.accession or b"-").decode(),
hit.evalue,
hit.score,
hit.bias,
diff --git a/pyhmmer/plan7.pyi b/pyhmmer/plan7.pyi
index 5c9fd058..4e8023c8 100644
--- a/pyhmmer/plan7.pyi
+++ b/pyhmmer/plan7.pyi
@@ -875,6 +875,8 @@ class TopHits(typing.Sequence[Hit]):
@property
def query_length(self) -> int: ...
@property
+ def query(self) -> typing.Union[DigitalSequence, DigitalMSA, HMM, Profile, OptimizedProfile]: ...
+ @property
def Z(self) -> float: ...
@property
def domZ(self) -> float: ...
--
GitLab